def test_merge_none():
pts = [
np.array([1, 1]),
np.array([0, 1]),
]
angles = [
math.pi / 4,
math.pi / 2,
]
seg1 = Segment(pts[0], pts[1], angles[0], angles[1])
seg1.merge(None)
np.testing.assert_allclose(seg1.p1, pts[0])
np.testing.assert_allclose(seg1.p2, pts[1])
assert seg1.theta1 == approx(angles[0])
assert seg1.theta2 == approx(angles[1])
def test_merge_before():
pts = [
np.array([1, 1]),
np.array([0, 1]),
np.array([-1, 1]),
]
angles = [
math.pi / 4,
math.pi / 2,
3 * math.pi / 4,
]
seg1 = Segment(pts[0], pts[1], angles[0], angles[1])
seg2 = Segment(pts[1], pts[2], angles[1], angles[2])
seg2.merge(seg1)
np.testing.assert_allclose(seg2.p1, pts[0])
np.testing.assert_allclose(seg2.p2, pts[2])
assert seg2.theta1 == approx(angles[0])
assert seg2.theta2 == approx(angles[2])
