def test_affine_jacobian_2d_with_positions():
params = np.array([0, 0.1, 0.2, 0, 30, 70])
t = AffineTransform.identity(2).from_vector(params)
explicit_pixel_locations = np.array(
[[0, 0],
[0, 1],
[0, 2],
[1, 0],
[1, 1],
[1, 2]])
dW_dp = t.jacobian(explicit_pixel_locations)
assert_equal(dW_dp, jac_solution2d)
def test_affine_jacobian_3d_with_positions():
params = np.ones(12)
t = AffineTransform.identity(3).from_vector(params)
explicit_pixel_locations = np.array(
[[0, 0, 0],
[0, 0, 1],
[0, 1, 0],
[0, 1, 1],
[0, 2, 0],
[0, 2, 1],
[1, 0, 0],
[1, 0, 1],
[1, 1, 0],
[1, 1, 1],
[1, 2, 0],
[1, 2, 1]])
dW_dp = t.jacobian(explicit_pixel_locations)
assert_equal(dW_dp, jac_solution3d)
def test_affine_identity_3d():
assert_allclose(AffineTransform.identity(3).h_matrix, np.eye(4))
