added test case

luxonis · Jan 14, 2025 · ac1ecab · ac1ecab
1 parent 728f1c9
commit ac1ecab
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_data/test_augmentations/test_batched.py b/tests/test_data/test_augmentations/test_batched.py
@@ -61,3 +61,17 @@ def test_mixup(
     config = [{"name": "MixUp", "params": {"p": 1.0}}]
     augmentations = AlbumentationsEngine(256, 256, targets, config)
     augmentations.apply([(image.copy(), deepcopy(labels)) for _ in range(2)])
+
+
+def test_batched_p_0(
+    image: np.ndarray, labels: Labels, targets: Dict[str, TaskType]
+):
+    config = [
+        {
+            "name": "Mosaic4",
+            "params": {"p": 0, "out_width": 640, "out_height": 640},
+        },
+        {"name": "MixUp", "params": {"p": 0}},
+    ]
+    augmentations = AlbumentationsEngine(256, 256, targets, config)
+    augmentations.apply([(image.copy(), deepcopy(labels)) for _ in range(8)])