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()
+