def test_polygon_new(self): polygon = PolygonNew() self.assertEqual(0, len(polygon.points)) polygon.pre_add_point(QPointF(2, 3)) self.assertEqual(0, len(polygon.points)) painter = mock.MagicMock() polygon.paint(painter, None) polygon.add_point(QPointF(4, 5)) self.assertEqual(1, len(polygon.points)) painter = mock.MagicMock() polygon.paint(painter, None) polygon.remove_point() self.assertEqual(0, len(polygon.points)) polygon.remove_point() self.assertEqual(0, len(polygon.points)) painter = mock.MagicMock() polygon.paint(painter, None)