From a2cdf719f710ce0e7c713872333ea2ec80b1d7e8 Mon Sep 17 00:00:00 2001 From: Caroline Malin-Mayor Date: Thu, 16 May 2024 16:48:41 +0200 Subject: [PATCH] Black test cases --- tests/cases/csv_points_source.py | 16 +++++++++------- tests/cases/shift_augment.py | 5 ++++- 2 files changed, 13 insertions(+), 8 deletions(-) diff --git a/tests/cases/csv_points_source.py b/tests/cases/csv_points_source.py index f53950b1..90c67cf9 100644 --- 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) \ No newline at end of file + + assert result_locs == unordered(target_locs) diff --git a/tests/cases/shift_augment.py b/tests/cases/shift_augment.py index f35862b6..b92c71f9 100644 --- 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))), )