added tests

luxonis · Nov 5, 2024 · cb823f8 · cb823f8
1 parent 5cb9b8d
commit cb823f8
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/test_data/test_dataset.py b/tests/test_data/test_dataset.py
@@ -572,3 +572,26 @@ def generator():
         assert "array" in labels
         assert labels["label"][0].tolist() == [["dog"], ["cat"]]
         assert np.allclose(labels["array"][0], arr)
+
+
+def test_no_labels():
+    dataset = LuxonisDataset("__no_labels", delete_existing=True)
+
+    def generator():
+        for i in range(10):
+            img = make_image(i)
+            yield {
+                "file": img,
+            }
+
+    dataset.add(generator())
+    dataset.make_splits()
+
+    loader = LuxonisLoader(dataset)
+    for _, labels in loader:
+        assert labels == {}
+
+    augments = Augmentations([512, 512], [{"name": "Flip", "params": {}}])
+    loader = LuxonisLoader(dataset, augmentations=augments)
+    for _, labels in loader:
+        assert labels == {}