Black test cases

funkelab · May 16, 2024 · a2cdf71 · a2cdf71
1 parent 7d57f92
commit a2cdf71
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/tests/cases/csv_points_source.py b/tests/cases/csv_points_source.py
@@ -13,9 +13,10 @@
     GraphSpec,
     build,
     Coordinate,
-    Roi
+    Roi,
 )
 
+
 # automatically set the seed for all tests
 @pytest.fixture(autouse=True)
 def seeds():
@@ -53,7 +54,10 @@ def test_pipeline3(test_points):
     csv_source = CsvPointsSource(
         fake_points_file,
         points_key,
-        spatial_cols=[0,1,],
+        spatial_cols=[
+            0,
+            1,
+        ],
         delimiter="\t",
         points_spec=GraphSpec(roi=Roi(shape=Coordinate((100, 100)), offset=(0, 0))),
     )
@@ -62,14 +66,12 @@ def test_pipeline3(test_points):
     shape = Coordinate((100, 100))
     request.add(points_key, shape)
 
-    pipeline = (
-        csv_source
-    )
+    pipeline = csv_source
     with build(pipeline) as b:
         request = b.request_batch(request)
 
     target_locs = [list(fake_point) for fake_point in fake_points]
     result_points = list(request[points_key].nodes)
     result_locs = [list(point.location) for point in result_points]
-    
-    assert result_locs == unordered(target_locs)
+
+    assert result_locs == unordered(target_locs)
diff --git a/tests/cases/shift_augment.py b/tests/cases/shift_augment.py
@@ -143,7 +143,10 @@ def test_pipeline3(test_points):
     csv_source = CsvPointsSource(
         fake_points_file,
         points_key,
-        spatial_cols=[0,1,],
+        spatial_cols=[
+            0,
+            1,
+        ],
         delimiter="\t",
         points_spec=GraphSpec(roi=Roi(shape=Coordinate((100, 100)), offset=(0, 0))),
     )