Added test for handling missing data using loadtxt.

1 files changed, 18 insertions, 0 deletions

diff --git a/numpy/lib/tests/test_io.py b/numpy/lib/tests/test_io.py
index c219f6fcd..d0187fb58 100644
--- a/numpy/lib/tests/test_io.py
+++ b/numpy/lib/tests/test_io.py
@@ -72,6 +72,24 @@ class Testloadtxt(NumpyTestCase):
         x = np.loadtxt(c, dtype=int)
         a = np.array([1,2,3,4], int)
         assert_array_equal(x, a)
+        
+        c = StringIO.StringIO()
+        c.write('1,2,3,4\n')
+        c.seek(0)
+        x = np.loadtxt(c, dtype=int, delimiter=',')
+        a = np.array([1,2,3,4], int)
+        assert_array_equal(x, a)
+        
+        
+    def test_missing(self):
+        c = StringIO.StringIO()
+        c.write('1,2,3,,5\n')
+        c.seek(0)
+        x = np.loadtxt(c, dtype=int, delimiter=',', \
+            converters={3:lambda s: int(s or -999)})
+        a = np.array([1,2,3,-999,5], int)
+        assert_array_equal(x, a)
+        
     
 if __name__ == "__main__":
     NumpyTest().run()