Add a test for no times given

dirac-institute · Feb 12, 2024 · 41bb952 · 41bb952
1 parent 7ca9059
commit 41bb952
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_fake_data_creator.py b/tests/test_fake_data_creator.py
@@ -37,6 +37,10 @@ def test_create(self):
             self.assertGreater(t, last_time)
             last_time = t
 
+    def test_create_empty_times(self):
+        ds = FakeDataSet(256, 128, [])
+        self.assertEqual(ds.stack.img_count(), 0)
+
     def test_insert_object(self):
         times = create_fake_times(5, 57130.2, 3, 0.01, 1)
         ds = FakeDataSet(128, 128, times, use_seed=True)