def manual_BasicStatsSeries_graph():
parent_dir, dir = tempdir()
bss = BasicStatsSeries(parent_directory=parent_dir, name=dir)
for i in range(50):
bss.append(1.0/numpy.arange(i*5, i*5+5))
bss.graph()
def test_SeriesList_withbasicstats():
dir = tempfile.mkdtemp(dir=BASEDIR)
bscstr = BasicStatsSeries.series_constructor()
slist = SeriesList(num_elements=5, name="foo", directory=dir, series_constructor=bscstr)
for i in range(10): # 10 elements in each list
curlist = []
for j in range(5): # 5 = num_elements, ie. number of list to append to
dist = numpy.arange(i*j, i*j+10)
curlist.append(dist)
slist.append(curlist)
slist2 = SeriesList(num_elements=5, name="foo", directory=dir, series_constructor=bscstr)
slist2.load_from_files()
l1 = slist2._subseries[0].means.tolist()
l2 = slist2._subseries[4].means.tolist()
print l1
print l2
assert l1 == [4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5]
assert l2 == [4.5, 8.5, 12.5, 16.5, 20.5, 24.5, 28.5, 32.5, 36.5, 40.5]
def test_BasicStatsSeries_withaccumulator():
a1 = numpy.arange(25).reshape((5,5))
a2 = numpy.arange(40).reshape((8,5))
a3 = numpy.arange(20).reshape((4,5))
a4 = numpy.arange(48).reshape((6,8))
parent_dir, dir = tempdir()
sc = AccumulatorSeries.series_constructor(reduce_every=2, mean=False)
bss = BasicStatsSeries(parent_directory=parent_dir, name=dir, series_constructor=sc)
bss.append(a1)
bss.append(a2)
bss.append(a3)
bss.append(a4)
assert bss.means.tolist() == [31.5, 33.0]
def test_BasicStatsSeries_commoncase():
a1 = numpy.arange(25).reshape((5,5))
a2 = numpy.arange(40).reshape((8,5))
parent_dir, dir = tempdir()
bss = BasicStatsSeries(parent_directory=parent_dir, name=dir)
bss.append(a1)
bss.append(a2)
assert bss.means.tolist() == [12.0, 19.5]
assert bss.mins.tolist() == [0.0, 0.0]
assert bss.maxes.tolist() == [24.0, 39.0]
assert (bss.stds.tolist()[0] - 7.211102) < 1e-3
assert (bss.stds.tolist()[1] - 11.54339) < 1e-3
# try to reload
bss2 = BasicStatsSeries(parent_directory=parent_dir, name=dir)
bss2.load_from_directory()
assert bss2.means.tolist() == [12.0, 19.5]
assert bss2.mins.tolist() == [0.0, 0.0]
assert bss2.maxes.tolist() == [24.0, 39.0]
assert (bss2.stds.tolist()[0] - 7.211102) < 1e-3
assert (bss2.stds.tolist()[1] - 11.54339) < 1e-3