def test_unsupported_suffix(self, tmpdir):
"""File with unsupported extension."""
# create a file with unsupported extensions
ext = "FAKE"
full_path = pathlib.Path(str(tmpdir)) / "test_file.{}".format(ext)
full_path.touch()
with pytest.raises(ValueError):
DisplacementField.from_file(full_path)
def test_npy(self, tmpdir, array, as_str):
"""Possible to read npy arrays."""
full_path = pathlib.Path(str(tmpdir)) / "file_temp.npy"
if as_str:
full_path = str(full_path)
np.save(full_path, array)
if array.ndim == 3 and array.shape[2] == 2:
df = DisplacementField.from_file(full_path)
assert isinstance(df, DisplacementField)
assert np.allclose(array[:, :, 0], df.delta_x)
assert np.allclose(array[:, :, 1], df.delta_y)
else:
with pytest.raises(ValueError):
DisplacementField.from_file(full_path)
def test_saveload(self, tmpdir, as_str):
"""Test that saving and loading loads the same dvf."""
delta_x = np.random.random((2, 3))
delta_y = np.random.random((2, 3))
df_orig = DisplacementField(delta_x, delta_y)
file_path = pathlib.Path(str(tmpdir)) / "test.npy"
if as_str:
file_path = str(file_path)
df_orig.save(file_path)
df_new = DisplacementField.from_file(file_path)
assert df_orig == df_new
def test_wrong_input(self):
"""Input is wrong."""
with pytest.raises(TypeError):
DisplacementField.from_file(123213)
