Skip to content

Commit c158e55

Browse files
committed
improve tests
1 parent ce69f51 commit c158e55

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

neo/test/rawiotest/test_intanrawio.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,10 @@ def test_correct_reading_one_file_per_channel(self):
7777

7878
# This should be the folder where the files of all the channels are stored
7979
folder_path = file_path.parent
80-
amplifier_file_paths = [p for p in folder_path.iterdir() if "amp" in p.name]
8180

82-
channel_names = [p.name[4:-4] for p in amplifier_file_paths]
81+
# The paths are named as amp-A-000.dat, amp-A-001.dat, amp-A-002.dat, ...
82+
amplifier_file_paths = [path for path in folder_path.iterdir() if "amp" in path.name]
83+
channel_names = [path.name[4:-4] for path in amplifier_file_paths]
8384

8485
for channel_name, amplifier_file_path in zip(channel_names, amplifier_file_paths):
8586
data_raw = np.fromfile(amplifier_file_path, dtype=np.int16).squeeze()

0 commit comments

Comments
 (0)