def test_load_matrix_flexible(tmpdir):
tiff_file = str(tmpdir.join("test.tiff"))
tif_file = str(tmpdir.join("test.tif"))
npy_file = str(tmpdir.join("test.npy"))
txt_file = str(tmpdir.join("test.txt"))
matrix = np.zeros([3, 3])
mod.write_tiff(tiff_file, matrix)
mod.write_tiff(tif_file, matrix)
np.save(npy_file, matrix)
np.savetxt(txt_file, matrix)
for f in [tiff_file, tif_file, npy_file, txt_file]:
matrix1 = mod.load_matrix_flexible(f)
assert np.array_equal(matrix1, matrix)
def test_load_matrix_flexible_error():
with pytest.raises(ValueError):
mod.load_matrix_flexible("test.jpg")
def user_mask(self):
mask_file = self.get("ANALYSIS", "user_mask", fallback="")
if not mask_file:
return None
return io.load_matrix_flexible(mask_file)
def user_mask(self) -> tp.Union[None, np.ndarray]:
"""The mask given by the user. Default None."""
mask_file = self._section.get(self._mask_file, fallback=None)
if mask_file:
return io.load_matrix_flexible(mask_file)
return None