@@ -93,8 +93,7 @@ def voltage_trace_verify(self, device):
93
93
for ref_vm , line in zip ((vm [::2 ], vm [1 ::2 ]), ax .lines ):
94
94
x_data , y_data = line .get_data ()
95
95
# Check that times are correct
96
- assert list (x_data ) == list (
97
- np .unique (device .get ("events" , "times" )))
96
+ assert list (x_data ) == list (np .unique (device .get ("events" , "times" )))
98
97
# Check that voltmeter data corresponds to the lines in the plot
99
98
assert all (np .isclose (ref_vm , y_data ))
100
99
plt .close (ax .get_figure ())
@@ -195,8 +194,7 @@ def test_raster_plot(self):
195
194
196
195
# Test extract_events
197
196
all_extracted = nest .raster_plot .extract_events (data )
198
- times_30_to_40_extracted = nest .raster_plot .extract_events (
199
- data , time = [30.0 , 40.0 ], sel = [3 ])
197
+ times_30_to_40_extracted = nest .raster_plot .extract_events (data , time = [30.0 , 40.0 ], sel = [3 ])
200
198
source_2_extracted = nest .raster_plot .extract_events (data , sel = [2 ])
201
199
assert np .array_equal (all_extracted , data )
202
200
assert np .all (times_30_to_40_extracted [:, 1 ] >= 30.0 )
0 commit comments