tests

Author: robertpfeiffer

test/draw_test.py

@@ -6311,6 +6311,45 @@ class to add any draw.arc specific tests to.
     """
 
 
+class DrawFloodFillMixin(unittest.TestCase):
+    """Mixin tests for flood fill.
+    """
+
+    def test_flood_fill(self):
+        """Ensures flood fill fills with solid color"""
+        surf = pygame.Surface((100, 100))
+        surf.fill((0,0,0))
+
+        pygame.draw.line (surf, (255,0,0), (10,10), (90,90), 5)
+
+        self.assertEqual(surface.get_at((10,10)), (255,0,0), "line drawing precondition")
+        self.assertEqual(surface.get_at((90,90)), (255,0,0), "line drawing precondition")
+
+        pygame.draw.flood_fill(surf, (255,255,255), (90, 90))
+
+        self.assertEqual(surface.get_at((90,90)), (255,255,255), "flood fill start point")
+        self.assertEqual(surface.get_at((10,10)), (255,255,255), "flood fill reaching the end")
+
+
+    def test_flood_pattern(self):
+        """Ensures flood fill fills in a pattern"""
+        surf = pygame.Surface((100, 100))
+        surf.fill((0,0,0))
+
+        pattern = pygame.Surface((2, 2))
+        pattern.fill((255,255,255))
+        pattern.set_at((0,0), (255,0,0))
+        pattern.set_at((1,1), (0,0,255))
+
+        pygame.draw.line (surf, (0,0,0), (5,95), (95,5))
+        pygame.draw.line (surf, (0,0,0), (50, 0), (50,95))
+
+        pygame.draw.flood_fill(surf, pattern, (95, 95))
+
+        for pt in [(0,0),(0,1),(1,0),(1,1)]:
+            self.assertEqual(surf.get_at(pt), pattern.get_at(pt), pt)
+
+
 ### Draw Module Testing #######################################################