49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
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")
|