test usecols

Author: legouee

neo/test/iotest/test_asciisignalio.py

@@ -155,7 +155,28 @@ def test_skiprows(self):
 
         os.remove(filename)
 
-    # test usecols
+    def test_usecols(self):
+        filename = 'test_usecols.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, usecols=3)
+        block = io.read_block()
+        signal = block.segments[0].analogsignals[0]
+        self.assertEqual(signal.shape, (5, 1))
+        self.assertEqual(len(block.segments[0].analogsignals[0]), 5)
+        self.assertEqual(len(block.segments[0].analogsignals), 1)
+        self.assertEqual(signal.units, pq.V)
+        
+        os.remove(filename)
+
 
     def test_timecolumn(self):
         sample_data = np.random.uniform(size=(200, 3))