tidy up, more tests

bugfixed datahandling.Field.exporttocsv.
tests added.

commit-on-a-plane :)

postpic/datahandling.py

@@ -93,7 +93,8 @@ def extent(self):
         if len(self._grid_node) < 2:
             ret = None
         else:
-            return [self._grid_node[0], self._grid_node[-1]]
+            ret = [self._grid_node[0], self._grid_node[-1]]
+        return ret
 
     @property
     def label(self):
@@ -366,10 +367,10 @@ def exporttocsv(self, filename):
         if self.dimensions == 1:
             data = np.asarray(self.matrix)
             x = np.linspace(self.extent[0], self.extent[1], len(data))
-            np.savetxt(dateiname, np.transpose([x, data]), delimiter=' ')
+            np.savetxt(filename, np.transpose([x, data]), delimiter=' ')
         elif self.dimensions == 2:
             export = np.asarray(self.matrix)
-            np.savetxt(dateiname, export)
+            np.savetxt(filename, export)
         else:
             raise Exception('Not Implemented')
         return

test/test_datahandling.py

@@ -90,6 +90,18 @@ def checkFieldConsistancy(self, field):
         for i in xrange(len(field.axes)):
             self.assertEqual(len(field.axes[i]), field.matrix.shape[i])
 
+    def test_extent(self):
+        self.assertListEqual(list(self.f0d.extent), [])
+        self.assertListEqual(list(self.f1d.extent), [0, 1])
+        self.f1d.extent = [3.3, 5.5]
+        self.assertListEqual(list(self.f1d.extent), [3.3, 5.5])
+        self.assertListEqual(list(self.f2d.extent), [0, 1, 0, 1])
+        self.f2d.extent = [3.3, 5.5, 7.7, 9.9]
+        self.assertListEqual(list(self.f2d.extent), [3.3, 5.5, 7.7, 9.9])
+        self.assertListEqual(list(self.f3d.extent), [0, 1, 0, 1, 0, 1])
+        self.f3d.extent = [3.3, 5.5, 7.7, 9.9, 11.1, 13.3]
+        self.assertListEqual(list(self.f3d.extent), [3.3, 5.5, 7.7, 9.9, 11.1,13.3])
+
     def test_dimensions(self):
         self.assertEqual(self.fempty.dimensions, -1)
         self.assertEqual(self.f0d.dimensions, 0)