def test__grid_are_again_the_same_after_centre_shift__grid_equivalent(self):
elliptical_profile1 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=55.0, centre=(1, 1)
)
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[1.0, 1.0]]))
)
elliptical_profile2 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=55.0, centre=(-1, -1)
)
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[-1.0, -1.0]]))
)
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__same_as_above_but_include_angle_phi_as_55__grid_are_equivalent(self):
elliptical_profile1 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=55.0, centre=(0, 0)
)
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[1.0, 1.0]]))
)
elliptical_profile2 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=55.0, centre=(-1, -1)
)
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[0.0, 0.0]]))
)
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__profile_angle_phi_is_60__cosine_and_sin_of_phi_follow_trig(self):
elliptical_profile = geometry_profiles.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 = geometry_profiles.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__grid_and_centres_of_two_lenses_are_equivalent__grid_are_equivalent(
self
):
elliptical_profile1 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=0.0, centre=(0, 0)
)
grid1 = elliptical_profile1.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[1.0, 1.0]]))
)
elliptical_profile2 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=0.0, centre=(-1, -1)
)
grid2 = elliptical_profile2.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(grid=np.array([[0.0, 0.0]]))
)
assert grid1[0, 0] == grid2[0, 0]
assert grid1[0, 1] == grid2[0, 1]
def test__profile_angle_phi_is_225__cosine_and_sin_of_phi_follow_trig__therefore_negative_half_root_2(
self
):
elliptical_profile = geometry_profiles.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__constructor_and_units(self):
profile = geometry_profiles.EllipticalProfile(
centre=(1.0, 2.0), axis_ratio=0.5, phi=45.0
)
assert profile.centre == (1.0, 2.0)
assert isinstance(profile.centre[0], aast.dim.Length)
assert isinstance(profile.centre[1], aast.dim.Length)
assert profile.centre[0].unit == "arcsec"
assert profile.centre[1].unit == "arcsec"
assert profile.axis_ratio == 0.5
assert isinstance(profile.axis_ratio, float)
assert profile.phi == 45.0
assert isinstance(profile.phi, float)
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 = geometry_profiles.EllipticalProfile(
axis_ratio=0.5, phi=0.0
)
eccentric_radius = elliptical_profile.grid_to_eccentric_radii(
aa.grid_irregular.manual_1d([[1.0, 1.0]])
)
assert eccentric_radius == pytest.approx(1.58113, 1e-3)
def test__profile_axis_ratio_1__r_is_root_2__therefore_ecccentric_radius_is_elliptical_radius_is_root_2(
self
):
elliptical_profile = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=0.0
)
eccentric_radius = elliptical_profile.grid_to_eccentric_radii(
aa.grid_irregular.manual_1d([[1.0, 1.0]])
)
assert eccentric_radius == pytest.approx(2.0 ** 0.5, 1e-3)
eccentric_radius = elliptical_profile.grid_to_eccentric_radii(
aa.coordinates([[(1.0, 1.0)]])
)
assert eccentric_radius[0][0] == pytest.approx(2.0 ** 0.5, 1e-3)
def test__profile_angle_phi_360__rotated_grid_are_original_grid_x_1_y_(self):
elliptical_profile = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=360.0
)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(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__random_values__grid_are_transformed_to_and_from_reference_frame__equal_to_original_values(
self
):
elliptical_profile = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=45.0
)
grid_original = aa.grid_irregular.manual_1d([[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 = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=90.0, centre=(2.0, 3.0)
)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(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)
def test__profile_angle_phi_315__grid_y_1_x_1__rotated_grid_clockwise_so_y_0_x_root_2(
self
):
elliptical_profile = geometry_profiles.EllipticalProfile(
axis_ratio=1.0, phi=315.0
)
transformed_grid = elliptical_profile.transform_grid_to_reference_frame(
grid=aa.grid_irregular.manual_1d(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)