diff _test_dataset.py @ 289:3af204aa71e5

implemented test for FieldsSubsetDataSet. if executed, can take a --debug option to be used in a debugger
author Frederic Bastien <bastienf@iro.umontreal.ca>
date Fri, 06 Jun 2008 16:15:43 -0400
parents 8e923cb2e8fc
children f7924e13e426
line wrap: on
line diff
--- a/_test_dataset.py	Fri Jun 06 14:08:28 2008 -0400
+++ b/_test_dataset.py	Fri Jun 06 16:15:43 2008 -0400
@@ -1,7 +1,7 @@
 #!/bin/env python
 from dataset import *
 from math import *
-import numpy,unittest
+import numpy, unittest, sys
 from misc import *
 
 def have_raised(to_eval, **var):
@@ -426,7 +426,13 @@
         del a,ds1,ds2,ds3
 
     def test_FieldsSubsetDataSet(self):
-        raise NotImplementedError()
+        a = numpy.random.rand(10,4)
+        ds = ArrayDataSet(a,LookupList(['x','y','z','w'],[slice(3),3,[0,2],0]))
+        ds = FieldsSubsetDataSet(ds,['x','y','z'])
+
+        test_all(a,ds)
+
+        del a, ds
     def test_MinibatchDataSet(self):
         raise NotImplementedError()
     def test_HStackedDataSet(self):
@@ -438,5 +444,14 @@
 
 
 if __name__=='__main__':
-    unittest.main()
+    if len(sys.argv)==2:
+        if sys.argv[1]=="--debug":
+            module = __import__("_test_dataset")
+            tests = unittest.TestLoader().loadTestsFromModule(module)
+            tests.debug()
+        print "bad argument: only --debug is accepted"
+    elif len(sys.argv)==1:
+        unittest.main()
+    else:
+        print "bad argument: only --debug is accepted"