def test_returns_unsigned_distances_for_xz_plane_at_origin(self):
# x-z plane
normal = np.array([0., 1., 0.])
sample = np.array([0., 0., 0.])
plane = Plane(sample, normal)
pts = np.array([
[500., 502., 503.],
[-500., -501., -503.],
])
expected = np.array([502., 501.])
np.testing.assert_array_equal(expected, plane.distance(pts))
def test_returns_unsigned_distances_for_diagonal_plane_at_origin(self):
# diagonal plane @ origin - draw a picture!
normal = np.array([1., 1., 0.])
normal /= np.linalg.norm(normal)
sample = np.array([0., 0., 0.])
plane = Plane(sample, normal)
pts = np.array([
[425., 425., 25.],
[-500., -500., 25.],
])
expected = np.array(
[math.sqrt(2 * (425.**2)),
math.sqrt(2 * (500.**2))])
np.testing.assert_array_almost_equal(expected, plane.distance(pts))