use get

Author: marqh

lib/bald/tests/__init__.pyc

@@ -154,16 +154,23 @@ def _check_ref(pdataset, parray, cdataset, carray):
                 exceptions.append(ValueError(msg))
             return exceptions
 
-        for attr, value in self.subject.attrs.iteritems():
-            # should support subtypes
-            if attr == 'bald__references':
-                # check if it's this type, otherwise exception)
-                # if isinstance(value,
-                child_dset = self.fhandle[value]
+        # if this Dataset has a bald__references
+        # not implemented yet: and it's a singleton
+        if self.subject.attrs.get('bald__references', ''):
+            # then it must have a bald_dataset
+            ref_dset = self.fhandle[self.subject.attrs.get('bald__references')]
+            if not ref_dset.attrs.get('bald__dataset', ''):
+                exceptions.append(ValueError('A bald__Reference must reference'
+                                             ' one and only one bald__Dataset')
+                                 )
+            else:
+                # and we impose bald broadcasting rules on it
+                child_dset = self.fhandle[ref_dset.attrs.get('bald__dataset')]
                 parray = np.zeros(self.dataset.shape)
                 carray = np.zeros(child_dset.shape)
 
                 exceptions = _check_ref('p', parray, 'c', carray)
+               
         return exceptions