def test_equality(self): p1 = Polygon(4, 10) p2 = Polygon(5, 10) p3 = Polygon(4, 11) self.assertEqual(self.poly, p1) self.assertNotEqual(self.poly, p2) self.assertNotEqual(self.poly, p3)
def test_poly_iterator(self): """ Iterator is being exhausted. """ iterator = self.polys.PolyIterator(5, 3) self.assertEqual(list(iterator), [Polygon(3, 3), Polygon(4, 3), Polygon(5, 3)]) self.assertEqual(list(iterator), [])
def test_iterable_polygons(self): """ Iterable can be iterated multiple times and is not exhausted. """ self.assertEqual(list(self.polys), [Polygon(3, 1), Polygon(4, 1), Polygon(5, 1)]) self.assertEqual(list(self.polys), [Polygon(3, 1), Polygon(4, 1), Polygon(5, 1)])
def setUp(self): self.poly = Polygon(4, 10)
def test_iteration(self): for idx, poly in enumerate(self.polys, 3): self.assertEqual(poly, Polygon(idx ,1))
def test_comparison(self): p1 = Polygon(3, 10) p2 = Polygon(5, 10) self.assertGreater(self.poly, p1) self.assertLess(self.poly, p2)
def test_polyons(self):
self.assertEqual(self.polys.polygons,
[Polygon(x, 1) for x in range(3, 6)])
def test_indexing(self):
self.assertEqual(self.polys[1], Polygon(4, 1))
self.assertEqual(self.polys[-1], Polygon(5, 1))
def test_slicing(self):
self.assertEqual(self.polys[1:], [Polygon(4, 1), Polygon(5, 1)])