Merge pull request #1483 from NeuralEnsemble/black-formatting

Black formatting

Author: zm711

neo/rawio/intanrawio.py

@@ -46,7 +46,7 @@ class IntanRawIO(BaseRawIO):
     ignore_integrity_checks: bool, default: False
         If True, data that violates integrity assumptions will be loaded. At the moment the only integrity
         check we perform is that timestamps are continuous. Setting this to True will ignore this check and set
-        the attribute `discontinuous_timestamps` to True if the timestamps are not continous. This attribute can be checked 
+        the attribute `discontinuous_timestamps` to True if the timestamps are not continous. This attribute can be checked
         after parsing the header to see if the timestamps are continuous or not.
     Notes
     -----
@@ -83,7 +83,7 @@ class IntanRawIO(BaseRawIO):
     11: 'Stim channel',
 
     * For the "header-attached" and "one-file-per-signal" formats, the structure of the digital input and output channels is
-    one long vector, which must be post-processed to extract individual digital channel information. 
+    one long vector, which must be post-processed to extract individual digital channel information.
     See the intantech website for more information on performing this post-processing.
 
     Examples
@@ -108,7 +108,6 @@ def __init__(self, filename="", ignore_integrity_checks=False):
         self.ignore_integrity_checks = ignore_integrity_checks
         self.discontinuous_timestamps = False
 
-
     def _source_name(self):
         return self.filename
 
@@ -207,7 +206,7 @@ def _parse_header(self):
         elif self.file_format == "one-file-per-channel":
             time_stream_index = max(self._raw_data.keys())
             timestamp = self._raw_data[time_stream_index][0]
-        
+
         discontinuous_timestamps = np.diff(timestamp) != 1
         timestamps_are_not_contiguous = np.any(discontinuous_timestamps)
         if timestamps_are_not_contiguous:
@@ -267,7 +266,7 @@ def _parse_header(self):
         # are in a list we just take the first channel in each list of channels
         else:
             self._max_sigs_length = max([raw_data[0].size for raw_data in self._raw_data.values()])
-            
+
         # No events
         event_channels = []
         event_channels = np.array(event_channels, dtype=_event_channel_dtype)

neo/test/iotest/test_intanio.py

@@ -15,13 +15,13 @@ class TestIntanIO(
     ioclass = IntanIO
     entities_to_download = ["intan"]
     entities_to_test = [
-        "intan/intan_rhs_test_1.rhs",  # Format header-attached 
+        "intan/intan_rhs_test_1.rhs",  # Format header-attached
         "intan/intan_rhd_test_1.rhd",  # Format header-attached
         "intan/rhs_fpc_multistim_240514_082243/rhs_fpc_multistim_240514_082243.rhs",  # Format header-attached newer version
         "intan/intan_fpc_test_231117_052630/info.rhd",  # Format one-file-per-channel
         "intan/intan_fps_test_231117_052500/info.rhd",  # Format one file per signal
         "intan/intan_fpc_rhs_test_240329_091637/info.rhs",  # Format one-file-per-channel
-        "intan/intan_fps_rhs_test_240329_091536/info.rhs",   # Format one-file-per-signal
+        "intan/intan_fps_rhs_test_240329_091536/info.rhs",  # Format one-file-per-signal
         "intan/rhd_fpc_multistim_240514_082044/info.rhd",  # Multiple digital channels one-file-per-channel rhd
     ]
 

neo/test/rawiotest/test_intanrawio.py

@@ -1,5 +1,5 @@
 import unittest
-import numpy as np 
+import numpy as np
 
 from neo.rawio.intanrawio import IntanRawIO
 from neo.test.rawiotest.common_rawio_test import BaseTestRawIO
@@ -12,32 +12,46 @@ class TestIntanRawIO(
     rawioclass = IntanRawIO
     entities_to_download = ["intan"]
     entities_to_test = [
-        "intan/intan_rhs_test_1.rhs",  # Format header-attached 
+        "intan/intan_rhs_test_1.rhs",  # Format header-attached
         "intan/intan_rhd_test_1.rhd",  # Format header-attached
         "intan/rhs_fpc_multistim_240514_082243/rhs_fpc_multistim_240514_082243.rhs",  # Format header-attached newer version
         "intan/intan_fpc_test_231117_052630/info.rhd",  # Format one-file-per-channel
         "intan/intan_fps_test_231117_052500/info.rhd",  # Format one file per signal
         "intan/intan_fpc_rhs_test_240329_091637/info.rhs",  # Format one-file-per-channel
-        "intan/intan_fps_rhs_test_240329_091536/info.rhs",   # Format one-file-per-signal
+        "intan/intan_fps_rhs_test_240329_091536/info.rhs",  # Format one-file-per-signal
         "intan/rhd_fpc_multistim_240514_082044/info.rhd",  # Multiple digital channels one-file-per-channel rhd
     ]
 
-
     def test_annotations(self):
-        
+
         intan_reader = IntanRawIO(filename=self.get_local_path("intan/intan_rhd_test_1.rhd"))
         intan_reader.parse_header()
-        
+
         raw_annotations = intan_reader.raw_annotations
         annotations = raw_annotations["blocks"][0]["segments"][0]  # Intan is mono segment
         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}
-        
+        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,
+        }
+
         for key in exepcted_annotations:
             if isinstance(exepcted_annotations[key], float):
                 self.assertAlmostEqual(signal_annotations[key], exepcted_annotations[key], places=2)
@@ -47,8 +61,11 @@ def test_annotations(self):
         # 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["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()