def test__change_border_mask__works_as_above(self):
thetas = np.linspace(0.0, 2.0 * np.pi, 16)
circle = list(map(lambda x: (np.cos(x), np.sin(x)), thetas))
coordinates = [(-2.0, 0.0), (-1.0, -1.0), (0.0, -2.0), (1.0, -1.0)] + circle + \
[(0.5, 0.0), (0.5, 0.5), (0.0, 0.5), (-0.5, 0.5)]
border_mask = [False] * 4 + [True] * 16 + [False] * 4
source_plane = analysis.SourcePlane(coordinates)
source_plane.relocate_coordinates_outside_border_with_mask_and_polynomial_degree(
border_mask, 3)
source_plane.sub_coordinates = map(
lambda r: pytest.approx(r, 1e-3), source_plane.sub_coordinates)
assert source_plane.sub_coordinates[:][0] == (-1.0, 0.0)
assert source_plane.sub_coordinates[:][1] == (-0.5 * math.sqrt(2),
-0.5 * math.sqrt(2))
assert source_plane.sub_coordinates[:][2] == (0.0, -1.0)
assert source_plane.sub_coordinates[:][3] == (0.5 * math.sqrt(2),
-0.5 * math.sqrt(2))
assert source_plane.sub_coordinates[:][4:24] == coordinates[:][
4:24]
def test__inside_and_outside_border_simple_cases__changes_where_appropriate(
self):
thetas = np.linspace(0.0, 2.0 * np.pi, 16)
circle = list(map(lambda x: (np.cos(x), np.sin(x)), thetas))
coordinates = circle + [(0.5, 0.0), (0.5, 0.5), (0.0, 0.5),
(-0.5, 0.5), (-2.0, 0.0), (-1.0, -1.0),
(0.0, -2.0), (1.0, -1.0)]
border_mask = [True] * 16 + [False] * 8
source_plane = analysis.SourcePlane(coordinates)
source_plane.relocate_coordinates_outside_border_with_mask_and_polynomial_degree(
border_mask, 3)
source_plane.sub_coordinates = map(
lambda r: pytest.approx(r, 1e-3), source_plane.sub_coordinates)
assert source_plane.sub_coordinates[:][0:20] == coordinates[:][
0:20]
assert source_plane.sub_coordinates[:][20] == (-1.0, 0.0)
assert source_plane.sub_coordinates[:][21] == (-0.5 * math.sqrt(2),
-0.5 * math.sqrt(2))
assert source_plane.sub_coordinates[:][22] == (0.0, -1.0)
assert source_plane.sub_coordinates[:][23] == (0.5 * math.sqrt(2),
-0.5 * math.sqrt(2))
def test__eight_coordinates_in_circle__thetas_at_each_coordinates_are_the_radius(
self):
coordinates = [(1.0, 0.0), (0.5 * math.sqrt(2),
0.5 * math.sqrt(2)), (0.0, 1.0),
(-0.5 * math.sqrt(2), 0.5 * math.sqrt(2)),
(-1.0, 0.0),
(-0.5 * math.sqrt(2), -0.5 * math.sqrt(2)),
(0.0, -1.0),
(0.5 * math.sqrt(2), -0.5 * math.sqrt(2))]
border_mask = [True, True, True, True, True, True, True, True]
source_plane = analysis.SourcePlane(coordinates)
border = source_plane.border_with_mask_and_polynomial_degree(
border_mask, 3)
assert border.border_radius_at_theta(theta=0.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=45.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=90.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=135.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=180.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=225.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=270.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=315.0) == pytest.approx(
1.0, 1e-3)
def test__x_and_y_coordinates_are_one__r_is_root_two(self):
coordinates = (1.0, 1.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
assert source_plane.coordinates_to_radius(
coordinates) == pytest.approx(np.sqrt(2), 1e-5)
def test__shift_x_and_y_coordinates_first__r_includes_shift(self):
coordinates = (3.0, 3.0)
source_plane = analysis.SourcePlane(coordinates, centre=(2.0, 2.0))
assert source_plane.coordinates_to_radius(
coordinates) == pytest.approx(math.sqrt(2.0), 1e-5)
def test__include_source_plane_centre__angle_takes_into_accounts(self):
coordinates = (2.0, 2.0)
source_plane = analysis.SourcePlane(coordinates, centre=(1.0, 1.0))
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == 45.0
def test__bottom_right_quandrant__angle_flips_back_to_above_90(self):
coordinates = (1.0, -1.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == 315.0
def test__bottom_left_quandrant__angle_continues_above_180(self):
coordinates = (-1.0, -1.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == 225.0
def test__top_left_quandrant__angle_goes_above_90(self):
coordinates = (-1.0, 1.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == 135.0
def test__angle_is_sixty__angles_follow_trig(self):
coordinates = (1.0, 1.7320)
source_plane = analysis.SourcePlane(coordinates)
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == pytest.approx(60.0, 1e-3)
def test__angle_is_forty_five__angles_follow_trig(self):
coordinates = (1.0, 1.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
theta_from_x = source_plane.coordinates_angle_from_x(coordinates)
assert theta_from_x == 45.0
def test__shift_x_coordinate_first__r_includes_shift(self):
coordinates = (1.0, 0.0)
source_plane = analysis.SourcePlane(coordinates,
centre=(-1.0, 0.0))
assert source_plane.coordinates_to_radius(
coordinates) == pytest.approx(2.0, 1e-5)
def test__source_plane_centre_x_and_y_shift__x_and_y_both_shift(self):
coordinates = (0.0, 0.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.5, 0.5))
coordinates_shift = source_plane.coordinates_to_centre(coordinates)
assert coordinates_shift[0] == -0.5
assert coordinates_shift[1] == -0.5
def test__source_plane_centre_and_coordinates__correct_shifts(self):
coordinates = (0.2, 0.4)
source_plane = analysis.SourcePlane(coordinates, centre=(1.0, 0.5))
coordinates_shift = source_plane.coordinates_to_centre(coordinates)
assert coordinates_shift[0] == -0.8
assert coordinates_shift[1] == pytest.approx(-0.1, 1e-5)
def test__source_plane_centre_zeros_by_default__no_shift(self):
coordinates = (0.0, 0.0)
source_plane = analysis.SourcePlane(coordinates)
coordinates_shift = source_plane.coordinates_to_centre(coordinates)
assert coordinates_shift[0] == 0.0
assert coordinates_shift[1] == 0.0
def test__source_plane_centre_x_shift__x_shifts(self):
coordinates = (0.0, 0.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.5, 0.0))
coordinates_shift = source_plane.coordinates_to_centre(
source_plane.sub_coordinates)
assert coordinates_shift[0] == -0.5
assert coordinates_shift[1] == 0.0
def test__four_coordinates_in_circle__correct_border(self):
coordinates = [(1.0, 0.0), (0.0, 1.0), (-1.0, 0.0), (0.0, -1.0)]
border_mask = [True, True, True, True]
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
border = source_plane.border_with_mask_and_polynomial_degree(
border_mask, 3)
assert border.coordinates == [(1.0, 0.0), (0.0, 1.0), (-1.0, 0.0),
(0.0, -1.0)]
assert border.radii == [1.0, 1.0, 1.0, 1.0]
assert border.thetas == [0.0, 90.0, 180.0, 270.0]
def test__source_plane_centre_offset__coordinates_same_r_and_theta_shifted(
self):
coordinates = [(2.0, 1.0), (1.0, 2.0), (0.0, 1.0), (1.0, 0.0)]
border_mask = [True, True, True, True]
source_plane = analysis.SourcePlane(coordinates, centre=(1.0, 1.0))
border = source_plane.border_with_mask_and_polynomial_degree(
border_mask, 3)
assert border.coordinates == [(2.0, 1.0), (1.0, 2.0), (0.0, 1.0),
(1.0, 0.0)]
assert border.radii == [1.0, 1.0, 1.0, 1.0]
assert border.thetas == [0.0, 90.0, 180.0, 270.0]
def test__test_other_thetas_radii(self):
coordinates = [(2.0, 0.0), (20., 20.), (2.0, 2.0), (-1.0, -1.0),
(0.0, -3.0), (1.0, 1.0)]
border_mask = [True, False, True, True, True, False]
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
border = source_plane.border_with_mask_and_polynomial_degree(
border_mask, 3)
assert border.coordinates == [(2.0, 0.0), (2.0, 2.0), (-1.0, -1.0),
(0.0, -3.0)]
assert border.radii == [
2.0, 2.0 * math.sqrt(2),
math.sqrt(2.0), 3.0
]
assert border.thetas == [0.0, 45.0, 225.0, 270.0]
def test__four_coordinates_in_circle__thetas_at_radius_are_each_coordinates_radius(
self):
coordinates = [(1.0, 0.0), (0.0, 1.0), (-1.0, 0.0), (0.0, -1.0),
(0.0, 0.0)]
border_mask = [True, True, True, True, False]
source_plane = analysis.SourcePlane(coordinates)
border = source_plane.border_with_mask_and_polynomial_degree(
border_mask, 3)
assert border.border_radius_at_theta(theta=0.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=90.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=180.0) == pytest.approx(
1.0, 1e-3)
assert border.border_radius_at_theta(theta=270.0) == pytest.approx(
1.0, 1e-3)
def test__x_coordinates_is_one__r_is_one(self):
coordinates = (1.0, 0.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
assert source_plane.coordinates_to_radius(coordinates) == 1.0
def test__coordinates_overlap_source_plane_analysis__r_is_zero(self):
coordinates = (0.0, 0.0)
source_plane = analysis.SourcePlane(coordinates, centre=(0.0, 0.0))
assert source_plane.coordinates_to_radius(coordinates) == 0.0
def test__sets_correct_values(self):
coordinates = [(1.0, 1.0), (0.0, 0.5)]
source_plane = analysis.SourcePlane(coordinates)
assert source_plane.sub_coordinates == [(1.0, 1.0), (0.0, 0.5)]
