test skiprows

Author: legouee

neo/test/iotest/test_asciisignalio.py

@@ -13,6 +13,7 @@
 from neo.test.iotest.common_io_test import BaseTestIO
 from neo.core import AnalogSignal, Segment, Block
 
+import string
 
 class TestAsciiSignalIOWithTestFiles(BaseTestIO, unittest.TestCase):
     ioclass = AsciiSignalIO
@@ -133,8 +134,27 @@ def reader(filename, comment_rows):
 
         os.remove(filename)
 
-    # test usecols
-    # test skiprows
+    def test_skiprows(self):
+        filename = 'test_skiprows.txt'
+        sample_data = [
+                (-65, -65, -65, 0.5),
+                (-64.8, -64.5, -64.0, 0.6),
+                (-64.6, -64.2, -77.0, 0.7),
+                (-64.3, -64.0, -99.9, 0.8),
+                (-66, -64.3, -61.0, 1.4),
+            ]
+        with open(filename, 'w') as datafile:
+            for row in sample_data:
+                datafile.write("\t ".join(map(str, row)) + "\t\n")
+        io = AsciiSignalIO(filename, skiprows=1)
+        block = io.read_block()
+        signal = block.segments[0].analogsignals[0]
+        self.assertEqual(signal.shape, (4, 1))
+        self.assertEqual(len(block.segments[0].analogsignals), 5)
+        self.assertEqual(len(block.segments[0].analogsignals[0]), 4)
+        self.assertEqual(signal.units, pq.V)
+
+        os.remove(filename)
 
     def test_timecolumn(self):
         sample_data = np.random.uniform(size=(200, 3))