Mercurial > pylearn
diff _test_dataset.py @ 220:1f527fe65e22
test on simple slicing works
author | Thierry Bertin-Mahieux <bertinmt@iro.umontreal.ca> |
---|---|
date | Fri, 23 May 2008 13:44:25 -0400 |
parents | 5b3afda2f1ad |
children | 58e17421c69c |
line wrap: on
line diff
--- a/_test_dataset.py Fri May 23 13:16:42 2008 -0400 +++ b/_test_dataset.py Fri May 23 13:44:25 2008 -0400 @@ -1,6 +1,8 @@ from dataset import * from math import * import unittest +import sys +import numpy as N def _sum_all(a): s=a @@ -92,10 +94,14 @@ print b('x+y') + + # to be used with a any new dataset class T_dataset_tester(object): """ This class' goal is to test any new dataset that is created + Tests are (will be!) designed to check the normal behaviours + of a dataset, as defined in dataset.py """ @@ -121,10 +127,20 @@ """test if slicing works properly""" print 'testing slicing...', sys.stdout.flush() + middle = len(ds) / 2 tenpercent = int(len(ds) * .1) set1 = ds[:middle+tenpercent] set2 = ds[middle-tenpercent:] + for k in range(tenpercent + tenpercent -1): + for k2 in ds.fieldNames() : + if type(set1[middle-tenpercent+k](k2)[0]) == N.ndarray : + for k3 in range(len(set1[middle-tenpercent+k](k2)[0])) : + assert set1[middle-tenpercent+k](k2)[0][k3] == set2[k](k2)[0][k3] + else : + assert set1[middle-tenpercent+k](k2)[0] == set2[k](k2)[0] + + print 'done' ###################################################################