changeset 81:4b0859606d05

Added test for ArrayDataSet and LookUpList
author Frederic Bastien <bastienf@iro.umontreal.ca>
date Mon, 05 May 2008 10:57:33 -0400
parents 40476a7746e8
children 158653a9bc7c
files test_dataset.py
diffstat 1 files changed, 76 insertions(+), 13 deletions(-) [+]
line wrap: on
line diff
--- a/test_dataset.py	Mon May 05 10:56:58 2008 -0400
+++ b/test_dataset.py	Mon May 05 10:57:33 2008 -0400
@@ -26,6 +26,9 @@
 def test_ArrayDataSet():
     #don't test stream
     #tested only with float value
+    #test with y too
+    #test missing value
+
     a = numpy.random.rand(10,4)
     print a
     ds = ArrayDataSet(a,{'x':slice(3),'y':3,'z':[0,2]})
@@ -36,17 +39,11 @@
         assert ds[i]['y']==a[i][3]
         assert ds[i]['z'].all()==a[i][0:3:2].all()
     print "x=",ds["x"]
-    print "x|y"
     i=0
     for x in ds('x','y'):
         assert numpy.append(x['x'],x['y']).all()==a[i].all()
         i+=1
-#    i=0
-#    for x in ds['x','y']: # don't work
-#        assert numpy.append(x['x'],x['y']).all()==a[i].all()
-#        i+=1
-#    for (x,y) in (ds('x','y'),a): #don't work # haven't found a variant that work.
-#        assert numpy.append(x,y)==z
+
     i=0
     for x,y in ds('x','y'):
         assert numpy.append(x,y).all()==a[i].all()
@@ -70,6 +67,14 @@
     except :
         have_thrown = True
     assert have_thrown == True
+
+    have_thrown = False
+    try:
+        ds[['h']] # h is not defined...
+    except :
+        have_thrown = True
+    assert have_thrown == True
+
     assert len(ds.fields())==3
     for field in ds.fields():
         for field_value in field: # iterate over the values associated to that field for all the ds examples
@@ -85,19 +90,29 @@
     assert ds == ds.fields().examples()
 
 
-    #test missing value
-
+    #ds[:n] returns a dataset with the n first examples.
     assert len(ds[:3])==3
     i=0
     for x,z in ds[:3]('x','z'):
         assert ds[i]['z'].all()==a[i][0:3:2].all()
         i+=1
+
     #ds[i1:i2:s]# returns a ds with the examples i1,i1+s,...i2-s.
-
-    #ds[i]# returns an Example.
-
+        ds[1:7:2][1] #fail???
+    assert len(ds[1:7:2])==3 # should be number example 1,3 and 5
+    i=0
+    for x,z in ds[1:7:2]('x','z'):
+        assert ds[i]['z'].all()==a[i][0:3:2].all()
+        i+=1
+    ds2=ds[1:7:2]
+    for i in range(len(ds2)):
+        print ds2[i]
     #ds[[i1,i2,...in]]# returns a ds with examples i1,i2,...in.
-
+    i=0
+    for x in ds[[1,2]]:
+        assert numpy.append(x['x'],x['y']).all()==a[i].all()
+        i+=1
+    #ds[i1,i2,...]# should we accept????
     #ds[fieldname]# an iterable over the values of the field fieldname across
       #the ds (the iterable is obtained by default by calling valuesVStack
       #over the values for individual examples).
@@ -110,5 +125,53 @@
     #* ds1 & ds2 & ds3 == ds.vstack([ds1,ds2,ds3])
 
 
+#    for (x,y) in (ds('x','y'),a): #don't work # haven't found a variant that work.
+#        assert numpy.append(x,y)==z
+
+def test_LookupList():
+    #test only the example in the doc???
+       example = LookupList(['x','y','z'],[1,2,3])
+       example['x'] = [1, 2, 3] # set or change a field
+       x, y, z = example
+       x = example[0]
+       x = example["x"]
+       assert example.keys()==['x','y','z']
+       assert example.values()==[[1,2,3],2,3]
+       assert example.items()==[('x',[1,2,3]),('y',2),('z',3)]
+       example.append_keyval('u',0) # adds item with name 'u' and value 0
+       assert len(example)==4 # number of items = 4 here
+       example2 = LookupList(['v','w'], ['a','b'])
+       example3 = LookupList(['x','y','z','u','v','w'], [[1, 2, 3],2,3,0,'a','b'])
+       print example3
+       print example+example2
+       print example+example2
+       assert example+example2==example3
+       have_throw=False
+       try:
+           example+example
+       except:
+           have_throw=True
+       assert have_throw
+
+def ApplyFunctionDataSet():
+    raise NotImplementedError()
+def CacheDataSet():
+    raise NotImplementedError()
+def FieldsSubsetDataSet():
+    raise NotImplementedError()
+def DataSetFields():
+    raise NotImplementedError()
+def MinibatchDataSet():
+    raise NotImplementedError()
+def HStackedDataSet():
+    raise NotImplementedError()
+def VStackedDataSet():
+    raise NotImplementedError()
+def ArrayFieldsDataSet():
+    raise NotImplementedError()
+
+test_LookupList()
 test_ArrayDataSet()
 
+
+