def test_get_distance_from_returns_positive_vals(self):
polygon = Polygon([[0, 0], [3, 3], [0, 3]])
piece1 = PolygonPiece(polygon, [1233, 12312])
self.assertTrue(
all(a > 0 for a in piece1.get_center_distance_from(0, 0)))
piece2 = PolygonPiece(polygon, [-1233, -12312])
self.assertTrue(
all(a > 0 for a in piece2.get_center_distance_from(0, 0)))
def test_get_center_distance_from_is_affected_by_polygon_piece_position(
self):
polygon = Polygon([[0, 0], [3, 3], [0, 3]])
piece1 = PolygonPiece(polygon, [0, 0])
piece2 = PolygonPiece(polygon, [15, 66])
p1pts = piece1.get_center_distance_from(0, 0)
p2pts = piece2.get_center_distance_from(0, 0)
self.assertFalse(all(a == b for a, b in zip(p1pts, p2pts)))
def test_get_center_distance_from_returns_0_if_centroid_in_same_point(
self):
polygon = Polygon([[0, 0], [3, 3], [0, 3]])
polygon.get_centroid_point = lambda: [1337, 1337]
piece = PolygonPiece(polygon, [0, 0])
distance = piece.get_center_distance_from(1337, 1337)
self.assertEqual(0, distance[0])
self.assertEqual(0, distance[1])
def test_get_center_distance_from_doesnt_return_0_if_centroid_in_different_point(
self):
polygon = Polygon([[0, 0], [3, 3], [0, 3]])
polygon.get_centroid_point = lambda: [51, 33]
piece = PolygonPiece(polygon, [0, 0])
distance = piece.get_center_distance_from(1337, 1337)
self.assertNotEqual(0, distance[0])
self.assertNotEqual(0, distance[1])
def test_get_center_distance_from_returns_array_with_two_numbers(self):
piece = PolygonPiece(Polygon([[0, 0], [3, 3], [0, 3]]), [125, 30])
distance = piece.get_center_distance_from(250, 300)
self.assertEqual(2, len(distance))