Mercurial > pylearn
diff _test_dataset.py @ 4:f7dcfb5f9d5b
Added test for dataset.
author | bengioy@bengiomac.local |
---|---|
date | Sun, 23 Mar 2008 22:14:10 -0400 |
parents | |
children | d5738b79089a |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/_test_dataset.py Sun Mar 23 22:14:10 2008 -0400 @@ -0,0 +1,25 @@ +from dataset import * +from math import * +import unittest + +def _sum_all(a): + s=a + while isinstance(s,numpy.ndarray): + s=sum(s) + return s + +class T_arraydataset(unittest.TestCase): + def setUp(self): + numpy.random.seed(123456) + + def test0(self): + a=ArrayDataSet(data=numpy.random.rand(8,3),fields={"x":slice(2),"y":slice(1,3)}) + s=0 + for example in a: + s+=_sum_all(example.x) + print s + self.failUnless(abs(s-11.4674133)<1e-6) + +if __name__ == '__main__': + unittest.main() +