Fixed missing variables and fixtures

Pale-Blue-Dot-97 · May 11, 2023 · 10c6dcb · 10c6dcb
1 parent 6c5f3d7
commit 10c6dcb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -117,7 +117,7 @@ def test_paired_datasets(img_root: Path) -> None:
     )
 
 
-def test_paired_union_datasets() -> None:
+def test_paired_union_datasets(img_root: Path) -> None:
     def dataset_test(_dataset) -> None:
         query_1 = get_random_bounding_box(bounds, (32, 32), 10.0)
         query_2 = get_random_bounding_box(bounds, (32, 32), 10.0)
@@ -126,6 +126,8 @@ def dataset_test(_dataset) -> None:
         assert type(sample_1) == dict
         assert type(sample_2) == dict
 
+    bounds = BoundingBox(411248.0, 412484.0, 4058102.0, 4059399.0, 0, 1e12)
+
     dataset1 = TstImgDataset(img_root)
     dataset2 = TstImgDataset(img_root)
     dataset3 = PairedDataset(TstImgDataset, img_root)