def test_process_seq_split(self):
self.sp = Splitter(self.output_folder)
self.sp.split_folders = True
self.sp.process(self.test_data_path)
assert os.path.exists(os.path.join(self.output_folder, "test", "raw"))
assert os.path.exists(
os.path.join(self.output_folder, "test", "preview"))
assert os.path.exists(
os.path.join(self.output_folder, "test", "radiometric"))
assert len(
glob.glob(os.path.join(self.output_folder, "test", "raw",
"*.fff"))) > 0
assert len(
glob.glob(os.path.join(self.output_folder, "test", "raw",
"*.txt"))) > 0
assert len(
glob.glob(
os.path.join(self.output_folder, "test", "preview",
"*.jpg"))) > 0
assert len(
glob.glob(
os.path.join(self.output_folder, "test", "radiometric",
"*.tiff"))) > 0
def test_raw_is_16_bit(self):
self.sp = Splitter(self.output_folder)
self.sp.process(self.test_data_path)
raw_files = glob.glob(
os.path.join(self.output_folder, "test", "radiometric", "*.tiff"))
for raw_file in raw_files:
assert (cv2.imread(raw_file,
cv2.IMREAD_UNCHANGED).dtype == "uint16")
def test_process_no_mmap(self):
self.sp = Splitter(self.output_folder)
self.sp.use_mmap = False
self.sp.process(self.test_data_path)
def test_output_folder(self):
self.sp = Splitter(self.output_folder)
assert os.path.exists(self.output_folder)
class TestSeqSplit:
def setup_method(self):
self.tempdir = tempfile.TemporaryDirectory()
self.output_folder = self.tempdir.name
folder = os.path.dirname(os.path.abspath(__file__))
self.test_data_path = os.path.join(folder, "test_data", "test.seq")
def teardown_method(self):
self.tempdir.cleanup()
def test_output_folder(self):
self.sp = Splitter(self.output_folder)
assert os.path.exists(self.output_folder)
def test_process_seq_split(self):
self.sp = Splitter(self.output_folder)
self.sp.split_folders = True
self.sp.process(self.test_data_path)
assert os.path.exists(os.path.join(self.output_folder, "test", "raw"))
assert os.path.exists(
os.path.join(self.output_folder, "test", "preview"))
assert os.path.exists(
os.path.join(self.output_folder, "test", "radiometric"))
assert len(
glob.glob(os.path.join(self.output_folder, "test", "raw",
"*.fff"))) > 0
assert len(
glob.glob(os.path.join(self.output_folder, "test", "raw",
"*.txt"))) > 0
assert len(
glob.glob(
os.path.join(self.output_folder, "test", "preview",
"*.jpg"))) > 0
assert len(
glob.glob(
os.path.join(self.output_folder, "test", "radiometric",
"*.tiff"))) > 0
def test_raw_is_16_bit(self):
self.sp = Splitter(self.output_folder)
self.sp.process(self.test_data_path)
raw_files = glob.glob(
os.path.join(self.output_folder, "test", "radiometric", "*.tiff"))
for raw_file in raw_files:
assert (cv2.imread(raw_file,
cv2.IMREAD_UNCHANGED).dtype == "uint16")
def test_process_no_mmap(self):
self.sp = Splitter(self.output_folder)
self.sp.use_mmap = False
self.sp.process(self.test_data_path)
def test_process_seq_no_split(self):
self.sp = Splitter(self.output_folder)
self.sp.split_filetypes = False
self.sp.process(self.test_data_path)
assert os.path.exists(os.path.join(self.output_folder, "test",
"raw")) is False
assert os.path.exists(
os.path.join(self.output_folder, "test", "preview")) is False
assert os.path.exists(
os.path.join(self.output_folder, "test", "radiometric")) is False
assert len(glob.glob(os.path.join(self.output_folder, "test",
"*.fff"))) > 0
assert len(glob.glob(os.path.join(self.output_folder, "test",
"*.txt"))) > 0
assert len(glob.glob(os.path.join(self.output_folder, "test",
"*.jpg"))) > 0
assert len(
glob.glob(os.path.join(self.output_folder, "test", "*.tiff"))) > 0
def test_process_with_wh(self):
self.sp = Splitter(self.output_folder, width=640, height=512)
self.sp.use_mmap = False
self.sp.process(self.test_data_path)