File: test_math_stuff.py

package info (click to toggle)
pytools 2014.3-2
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 424 kB
  • ctags: 971
  • sloc: python: 4,037; makefile: 14
file content (23 lines) | stat: -rw-r--r-- 491 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
from __future__ import division




def test_variance():
    data = [4, 7, 13, 16]

    def naive_var(data):
        n = len(data)
        return ((
            sum(di**2 for di in data)
            - sum(data)**2/n)
            /(n-1))

    from pytools import variance
    orig_variance = variance(data, entire_pop=False)

    assert abs(naive_var(data) - orig_variance) < 1e-15

    data = [1e9 + x for x in data]
    assert abs(variance(data, entire_pop=False) - orig_variance) < 1e-15