Fix Black's stage erros

nest · Feb 10, 2025 · 154d082 · 154d082
1 parent 6520b06
commit 154d082
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/testsuite/pytests/test_visualization.py b/testsuite/pytests/test_visualization.py
@@ -93,8 +93,7 @@ def voltage_trace_verify(self, device):
         for ref_vm, line in zip((vm[::2], vm[1::2]), ax.lines):
             x_data, y_data = line.get_data()
             # Check that times are correct
-            assert list(x_data) == list(
-                np.unique(device.get("events", "times")))
+            assert list(x_data) == list(np.unique(device.get("events", "times")))
             # Check that voltmeter data corresponds to the lines in the plot
             assert all(np.isclose(ref_vm, y_data))
         plt.close(ax.get_figure())
@@ -195,8 +194,7 @@ def test_raster_plot(self):
 
         # Test extract_events
         all_extracted = nest.raster_plot.extract_events(data)
-        times_30_to_40_extracted = nest.raster_plot.extract_events(
-            data, time=[30.0, 40.0], sel=[3])
+        times_30_to_40_extracted = nest.raster_plot.extract_events(data, time=[30.0, 40.0], sel=[3])
         source_2_extracted = nest.raster_plot.extract_events(data, sel=[2])
         assert np.array_equal(all_extracted, data)
         assert np.all(times_30_to_40_extracted[:, 1] >= 30.0)