def test_noisy(self):
shape = (10, 11)
delta_x = np.random.randint(1, 20, size=shape)
delta_y = np.random.randint(1, 20, size=shape)
df = DisplacementField(delta_x, delta_y)
df_anchored = df.anchor(ds_f=1)
assert df != df_anchored
def test_interpolation_subgrid(self):
shape = (14, 16) # center is 7, 8
eps = 1e-6
delta_x = np.random.randint(1, 20, size=shape)
delta_y = np.random.randint(1, 20, size=shape)
df = DisplacementField(delta_x, delta_y)
df_anchored_interpolated = df.anchor(h_kept=4,
w_kept=4,
ds_f=1,
smooth=0)
df_anchored_smoothened = df.anchor(h_kept=4,
w_kept=4,
ds_f=1,
smooth=1)
assert np.allclose(df_anchored_interpolated.delta_x[5:9, 6:10],
df.delta_x[5:9, 6:10])
assert np.allclose(df_anchored_interpolated.delta_y[5:9, 6:10],
df.delta_y[5:9, 6:10])
assert not np.allclose(df_anchored_smoothened.delta_x[5:9, 6:10],
df.delta_x[5:9, 6:10])
assert not np.allclose(df_anchored_smoothened.delta_y[5:9, 6:10],
df.delta_y[5:9, 6:10])
assert abs(df_anchored_interpolated.delta_x[0, 0]) < eps
assert abs(df_anchored_interpolated.delta_x[0, -1]) < eps
assert abs(df_anchored_interpolated.delta_x[-1, 0]) < eps
assert abs(df_anchored_interpolated.delta_x[-1, -1]) < eps
assert abs(df_anchored_interpolated.delta_y[0, 0]) < eps
assert abs(df_anchored_interpolated.delta_y[0, -1]) < eps
assert abs(df_anchored_interpolated.delta_y[-1, 0]) < eps
assert abs(df_anchored_interpolated.delta_y[-1, -1]) < eps