feat: init

test/model/features/test_image_writer.py

@@ -0,0 +1,48 @@
+import os
+import pytest
+
+from navigate.tools.file_functions import delete_folder
+
+
+@pytest.fixture()
+def image_writer(dummy_model):
+    from navigate.model.features.image_writer import ImageWriter
+
+    model = dummy_model
+    model.configuration["experiment"]["Saving"]["save_directory"] = "test_save_dir"
+
+    writer = ImageWriter(dummy_model)
+
+    yield writer
+
+    writer.close()
+
+
+def test_image_write_fail(image_writer):
+    image_writer.save_image([-1, image_writer.model.data_buffer.shape[0]])
+
+    # make sure the directory is empty
+    ls = os.listdir("test_save_dir")
+    ls.remove("MIP")
+    assert not ls
+
+    delete_folder("test_save_dir")
+
+
+def test_image_write(image_writer):
+    from numpy.random import rand
+
+    # Randomize the data buffer
+    for i in range(image_writer.model.data_buffer.shape[0]):
+        image_writer.model.data_buffer[i, ...] = rand(
+            image_writer.model.img_width, image_writer.model.img_height
+        )
+
+    image_writer.save_image(list(range(image_writer.model.number_of_frames)))
+
+    # make sure the directory isn't empty
+    ls = os.listdir("test_save_dir")
+    ls.remove("MIP")
+    assert ls
+
+    delete_folder("test_save_dir")