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