add some scalar annotations

Author: h-mayorquin

neo/test/rawiotest/test_intanrawio.py

@@ -1,7 +1,7 @@
 import unittest
+import numpy as np 
 
 from neo.rawio.intanrawio import IntanRawIO
-
 from neo.test.rawiotest.common_rawio_test import BaseTestRawIO
 
 
@@ -31,6 +31,7 @@ def test_annotations(self):
         signal_annotations = annotations["signals"][0]  # As in the other exmaples, annotaions are duplicated
 
 
+        # Scalar annotations
         exepcted_annotations =  {'intan_version': '1.5', 'desired_impedance_test_frequency': 1000.0, 'desired_upper_bandwidth': 7500.0, 'note1': '', 'notch_filter_mode': 1, 'notch_filter': False, 'nb_signal_group': 7, 
                             'dsp_enabled': 1, 'actual_impedance_test_frequency': 1000.0, 'desired_lower_bandwidth': 0.1, 'note3': '', 'actual_dsp_cutoff_frequency': 1.165828, 
                             'desired_dsp_cutoff_frequency': 1.0, 'actual_lower_bandwidth': 0.0945291, 'eval_board_mode': 0, 'note2': '', 'num_temp_sensor_channels': 0}
@@ -41,6 +42,10 @@ def test_annotations(self):
             else:
                 self.assertEqual(signal_annotations[key], exepcted_annotations[key])
 
-   
+        # Array annotations
+        signal_array_annotations = signal_annotations["__array_annotations__"]
+        np.testing.assert_array_equal(signal_array_annotations["native_order"][:10], [0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
+        np.testing.assert_array_equal(signal_array_annotations["spike_scope_digital_edge_polarity"][:10], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
+        np.testing.assert_array_equal(signal_array_annotations["board_stream_num"][:10], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
 if __name__ == "__main__":
     unittest.main()