def test__same_as_above_but_include_angle_phi_as_55__grid_are_equivalent(
self):
elliptical_profile1 = gp.EllipticalProfile(axis_ratio=1.0,
phi=55.0,
centre=(0, 0))
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
elliptical_profile2 = gp.EllipticalProfile(axis_ratio=1.0,
phi=55.0,
centre=(-1, -1))
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=np.array([[0.0, 0.0]]))
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__grid_are_again_the_same_after_centre_shift__grid_equivalent(
self):
elliptical_profile1 = gp.EllipticalProfile(axis_ratio=1.0,
phi=55.0,
centre=(1, 1))
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
elliptical_profile2 = gp.EllipticalProfile(axis_ratio=1.0,
phi=55.0,
centre=(-1, -1))
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=np.array([[-1.0, -1.0]]))
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__grid_and_centres_of_two_lenses_are_equivalent__grid_are_equivalent(
self):
elliptical_profile1 = gp.EllipticalProfile(axis_ratio=1.0,
phi=0.0,
centre=(0, 0))
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
elliptical_profile2 = gp.EllipticalProfile(axis_ratio=1.0,
phi=0.0,
centre=(-1, -1))
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=np.array([[0.0, 0.0]]))
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__profile_axis_ratio_1__r_is_root_2__therefore_ecccentric_radius_is_elliptical_radius_is_root_2(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0, phi=0.0)
eccentric_radius = elliptical_profile.grid_to_eccentric_radii(
np.array([[1.0, 1.0]]))
assert eccentric_radius == pytest.approx(2.0**0.5, 1e-3)
def test__profile_angle_phi_is_225__cosine_and_sin_of_phi_follow_trig__therefore_negative_half_root_2(
self):
elliptical_profile = gp.EllipticalProfile(centre=(1, 1),
axis_ratio=1.0,
phi=225.0)
cos_phi, sin_phi = elliptical_profile.cos_and_sin_from_x_axis()
assert cos_phi == pytest.approx(-0.707, 1e-3)
assert sin_phi == pytest.approx(-0.707, 1e-3)
def test__profile_angle_phi_is_60__cosine_and_sin_of_phi_follow_trig(
self):
elliptical_profile = gp.EllipticalProfile(centre=(1, 1),
axis_ratio=1.0,
phi=60.0)
cos_phi, sin_phi = elliptical_profile.cos_and_sin_from_x_axis()
assert cos_phi == pytest.approx(0.5, 1e-3)
assert sin_phi == pytest.approx(0.866, 1e-3)
def test__profile_angle_phi_is_0__cosine_and_sin_of_phi_is_1_and_0(
self):
elliptical_profile = gp.EllipticalProfile(centre=(1.0, 1.0),
axis_ratio=1.0,
phi=0.0)
cos_phi, sin_phi = elliptical_profile.cos_and_sin_from_x_axis()
assert cos_phi == 1.0
assert sin_phi == 0.0
def test__same_as_above_but_axis_ratio_is_05__r_follows_elliptical_coordinate_equation(
self):
# eccentric_radius = sqrt(axis_ratio) * sqrt( x**2 + (y**2 / axis_ratio**2))
# eccentric_radius = sqrt(0.5) * sqrt(1.0**2 + (1.0**2 / 0.5**2))
# eccentric radius = sqrt(0.5) * sqrt( 5 ) = 1.58113
elliptical_profile = gp.EllipticalProfile(axis_ratio=0.5, phi=0.0)
eccentric_radius = elliptical_profile.grid_to_eccentric_radii(
np.array([[1.0, 1.0]]))
assert eccentric_radius == pytest.approx(1.58113, 1e-3)
def test__profile_angle_phi_90__grid_y_1_x_1__rotated_grid_clockwise_so_y_1_x_negative_1(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0, phi=90.0)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
assert transformed_grid[0, 0] == pytest.approx(-1.0, 1e-3)
assert transformed_grid[0, 1] == pytest.approx(1.0, 1e-3)
transformed_back_grid = elliptical_profile.transform_grid_from_reference_frame(
transformed_grid)
assert transformed_back_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_back_grid[0, 1] == pytest.approx(1.0, 1e-3)
def test__profile_angle_phi_is_0__grid_x_1_y_1__returns_same_grid_so_x_1_y_1(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0, phi=0.0)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
assert transformed_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_grid[0, 1] == pytest.approx(1.0, 1e-3)
transformed_back_grid = elliptical_profile.transform_grid_from_reference_frame(
transformed_grid)
assert transformed_back_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_back_grid[0, 1] == pytest.approx(1.0, 1e-3)
def test__profile_angle_phi_360__rotated_grid_are_original_grid_x_1_y_(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0,
phi=360.0)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
assert transformed_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_grid[0, 1] == pytest.approx(1.0, 1e-3)
transformed_back_grid = elliptical_profile.transform_grid_from_reference_frame(
transformed_grid)
assert transformed_back_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_back_grid[0, 1] == pytest.approx(1.0, 1e-3)
def test___profile_angle_phi_45__grid_y_1_x_1__rotated_counter_clockwise_to_y_0_x_root_2(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0,
phi=315.0)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=np.array([[1.0, 1.0]]))
assert transformed_grid[0, 0] == pytest.approx(2**0.5, 1e-3)
assert transformed_grid[0, 1] == pytest.approx(0.0, 1e-3)
transformed_back_grid = elliptical_profile.transform_grid_from_reference_frame(
transformed_grid)
assert transformed_back_grid[0, 0] == pytest.approx(1.0, 1e-3)
assert transformed_back_grid[0, 1] == pytest.approx(1.0, 1e-3)
def test__random_values__grid_are_transformed_to_and_from_reference_frame__equal_to_original_values(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0, phi=45.0)
grid_original = np.array([[5.2221, 2.6565]])
grid_elliptical = elliptical_profile.transform_grid_to_reference_frame(
grid_original)
transformed_grid = elliptical_profile.transform_grid_from_reference_frame(
grid_elliptical)
assert transformed_grid[0, 0] == pytest.approx(
grid_original[0, 0], 1e-5)
assert transformed_grid[0, 1] == pytest.approx(
grid_original[0, 1], 1e-5)
def test__include_profile_centre_offset__is_used_before_rotation_is_performed(
self):
elliptical_profile = gp.EllipticalProfile(axis_ratio=1.0,
phi=90.0,
centre=(2.0, 3.0))
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=np.array([[3.0, 4.0]]))
assert transformed_grid[0, 0] == pytest.approx(-1.0, 1e-3)
assert transformed_grid[0, 1] == pytest.approx(1.0, 1e-3)
transformed_back_grid = elliptical_profile.transform_grid_from_reference_frame(
transformed_grid)
assert transformed_back_grid[0, 0] == pytest.approx(3.0, 1e-3)
assert transformed_back_grid[0, 1] == pytest.approx(4.0, 1e-3)