diff _test_dataset.py @ 29:46c5c90019c2

Changed apply_function so that it propagates methods of the source.
author bengioy@grenat.iro.umontreal.ca
date Fri, 11 Apr 2008 15:46:18 -0400
parents 541a273bc89f
children 5b3afda2f1ad
line wrap: on
line diff
--- a/_test_dataset.py	Fri Apr 11 13:08:51 2008 -0400
+++ b/_test_dataset.py	Fri Apr 11 15:46:18 2008 -0400
@@ -80,6 +80,17 @@
         b=a.rename({'xx':'x','zz':'z'})
         self.failUnless(b.hasFields('xx','zz') and not b.hasFields('x') and not b.hasFields('y'))
 
+class T_applyfunctiondataset(unittest.TestCase):
+    def setUp(self):
+        numpy.random.seed(123456)
+
+    def test_function(self):
+        n = numpy.random.rand(3,8)
+        a=ArrayDataSet(data=n,fields={"x":slice(2),"y":slice(1,4),"z":slice(4,6)})
+        b=a.apply_function(lambda x,y: x+y,x+1, ['x','y'], ['x+y','x+1'], False,False,False)
+        print b.fieldNames()
+        print b('x+y')
+        
 
 if __name__ == '__main__':
     unittest.main()