def test_nearest_with_arrays():
angle = np.deg2rad(45.)
pts = nearest_point_on_ray(((1., ), (0., )), (0., 0.), angle)
assert_array_almost_equal(pts, [0.5, 0.5])
pts = nearest_point_on_ray(((1., 0.), (0., 2.)), (0., 0.), angle)
assert_array_almost_equal(pts, [[0.5, 1.0], [0.5, 1.0]])
def test_nearest_to_y_axis():
"""Test if the ray is the +- y-axis."""
for angle in (np.pi / 2., 5. * np.pi / 2.):
pts = nearest_point_on_ray(((1., -1.), (1., -1.)), angle=angle)
assert_array_almost_equal(pts, [[0., 0.], [1., 0.]])
for angle in (-np.pi / 2., 3. * np.pi / 2.):
pts = nearest_point_on_ray(((1., -1.), (1., -1.)), angle=angle)
assert_array_almost_equal(pts, [[0., 0.], [0., -1.]])
def test_nearest_to_x_axis():
"""Test if the ray is the +- x-axis."""
for angle in (0., 2 * np.pi):
pts = nearest_point_on_ray(((1., -1.), (1., -1.)), angle=angle)
assert_array_almost_equal(pts, [[1., 0.], [0., 0.]])
for angle in (-np.pi, np.pi):
pts = nearest_point_on_ray(((1., -1.), (1., -1.)), angle=angle)
assert_array_almost_equal(pts, [[0., -1.], [0., 0.]])
def test_nearest_behind_origin():
"""Test if the nearest point is the origin."""
pts = nearest_point_on_ray((-1., 0.))
assert_array_almost_equal(pts, [0., 0.])
def test_nearest_on_line():
"""Test if the point is already on the line."""
pts = nearest_point_on_ray(((1., 0.), (1., 0.)))
assert_array_almost_equal(pts, [[1., 0.], [1., 0.]])
def test_nearest_with_defaults():
"""Test with default optional arguments."""
pts = nearest_point_on_ray(((0., 3.), (0., 1.)))
assert_array_almost_equal(pts, [[0., 2.], [0., 2.]])
def test_nearest_changing_angle():
"""Test using the angle keyword."""
pts = nearest_point_on_ray(((1., 1., 0.), (1., -1., -10)),
angle=-np.pi / 4.)
assert_array_almost_equal(pts, [[0., 1., 5.], [0., -1., -5.]])
def test_nearest_moving_origin():
"""Test using the origin keyword."""
pts = nearest_point_on_ray(((1., 1., 1.), (1., 2., 10.)), origin=(1., 2.))
assert_array_almost_equal(pts, [[1., 1., 5.], [2., 2., 6.]])
def test_nearest_with_scalars():
"""Test passing scalars."""
pts = nearest_point_on_ray((1., 0.), (0., 0.), np.deg2rad(45.))
assert_array_almost_equal(pts, [0.5, 0.5])