import math import unittest import sys import warnings import pygame from pygame import draw from pygame import draw_py from pygame.locals import SRCALPHA from pygame.tests import test_utils from pygame.math import Vector2 RED = BG_RED = pygame.Color("red") GREEN = FG_GREEN = pygame.Color("green") # Clockwise from the top left corner and ending with the center point. RECT_POSITION_ATTRIBUTES = ( "topleft", "midtop", "topright", "midright", "bottomright", "midbottom", "bottomleft", "midleft", "center", ) def get_border_values(surface, width, height): """Returns a list containing lists with the values of the surface's borders. """ border_top = [surface.get_at((x, 0)) for x in range(width)] border_left = [surface.get_at((0, y)) for y in range(height)] border_right = [surface.get_at((width - 1, y)) for y in range(height)] border_bottom = [surface.get_at((x, height - 1)) for x in range(width)] return [border_top, border_left, border_right, border_bottom] def corners(surface): """Returns a tuple with the corner positions of the given surface. Clockwise from the top left corner. """ width, height = surface.get_size() return ((0, 0), (width - 1, 0), (width - 1, height - 1), (0, height - 1)) def rect_corners_mids_and_center(rect): """Returns a tuple with each corner, mid, and the center for a given rect. Clockwise from the top left corner and ending with the center point. """ return ( rect.topleft, rect.midtop, rect.topright, rect.midright, rect.bottomright, rect.midbottom, rect.bottomleft, rect.midleft, rect.center, ) def border_pos_and_color(surface): """Yields each border position and its color for a given surface. Clockwise from the top left corner. """ width, height = surface.get_size() right, bottom = width - 1, height - 1 # Top edge. for x in range(width): pos = (x, 0) yield pos, surface.get_at(pos) # Right edge. # Top right done in top edge loop. for y in range(1, height): pos = (right, y) yield pos, surface.get_at(pos) # Bottom edge. # Bottom right done in right edge loop. for x in range(right - 1, -1, -1): pos = (x, bottom) yield pos, surface.get_at(pos) # Left edge. # Bottom left done in bottom edge loop. Top left done in top edge loop. for y in range(bottom - 1, 0, -1): pos = (0, y) yield pos, surface.get_at(pos) def get_color_points(surface, color, bounds_rect=None, match_color=True): """Get all the points of a given color on the surface within the given bounds. If bounds_rect is None the full surface is checked. If match_color is True, all points matching the color are returned, otherwise all points not matching the color are returned. """ get_at = surface.get_at # For possible speed up. if bounds_rect is None: x_range = range(surface.get_width()) y_range = range(surface.get_height()) else: x_range = range(bounds_rect.left, bounds_rect.right) y_range = range(bounds_rect.top, bounds_rect.bottom) surface.lock() # For possible speed up. if match_color: pts = [(x, y) for x in x_range for y in y_range if get_at((x, y)) == color] else: pts = [(x, y) for x in x_range for y in y_range if get_at((x, y)) != color] surface.unlock() return pts def create_bounding_rect(surface, surf_color, default_pos): """Create a rect to bound all the pixels that don't match surf_color. The default_pos parameter is used to position the bounding rect for the case where all pixels match the surf_color. """ width, height = surface.get_clip().size xmin, ymin = width, height xmax, ymax = -1, -1 get_at = surface.get_at # For possible speed up. surface.lock() # For possible speed up. for y in range(height): for x in range(width): if get_at((x, y)) != surf_color: xmin = min(x, xmin) xmax = max(x, xmax) ymin = min(y, ymin) ymax = max(y, ymax) surface.unlock() if -1 == xmax: # No points means a 0 sized rect positioned at default_pos. return pygame.Rect(default_pos, (0, 0)) return pygame.Rect((xmin, ymin), (xmax - xmin + 1, ymax - ymin + 1)) class InvalidBool: """To help test invalid bool values.""" __bool__ = None class DrawTestCase(unittest.TestCase): """Base class to test draw module functions.""" draw_rect = staticmethod(draw.rect) draw_polygon = staticmethod(draw.polygon) draw_circle = staticmethod(draw.circle) draw_ellipse = staticmethod(draw.ellipse) draw_arc = staticmethod(draw.arc) draw_line = staticmethod(draw.line) draw_lines = staticmethod(draw.lines) draw_aaline = staticmethod(draw.aaline) draw_aalines = staticmethod(draw.aalines) class PythonDrawTestCase(unittest.TestCase): """Base class to test draw_py module functions.""" # draw_py is currently missing some functions. # draw_rect = staticmethod(draw_py.draw_rect) draw_polygon = staticmethod(draw_py.draw_polygon) # draw_circle = staticmethod(draw_py.draw_circle) # draw_ellipse = staticmethod(draw_py.draw_ellipse) # draw_arc = staticmethod(draw_py.draw_arc) draw_line = staticmethod(draw_py.draw_line) draw_lines = staticmethod(draw_py.draw_lines) draw_aaline = staticmethod(draw_py.draw_aaline) draw_aalines = staticmethod(draw_py.draw_aalines) ### Ellipse Testing ########################################################### class DrawEllipseMixin: """Mixin tests for drawing ellipses. This class contains all the general ellipse drawing tests. """ def test_ellipse__args(self): """Ensures draw ellipse accepts the correct args.""" bounds_rect = self.draw_ellipse( pygame.Surface((3, 3)), (0, 10, 0, 50), pygame.Rect((0, 0), (3, 2)), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__args_without_width(self): """Ensures draw ellipse accepts the args without a width.""" bounds_rect = self.draw_ellipse( pygame.Surface((2, 2)), (1, 1, 1, 99), pygame.Rect((1, 1), (1, 1)) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__args_with_negative_width(self): """Ensures draw ellipse accepts the args with negative width.""" bounds_rect = self.draw_ellipse( pygame.Surface((3, 3)), (0, 10, 0, 50), pygame.Rect((2, 3), (3, 2)), -1 ) self.assertIsInstance(bounds_rect, pygame.Rect) self.assertEqual(bounds_rect, pygame.Rect(2, 3, 0, 0)) def test_ellipse__args_with_width_gt_radius(self): """Ensures draw ellipse accepts the args with width > rect.w // 2 and width > rect.h // 2. """ rect = pygame.Rect((0, 0), (4, 4)) bounds_rect = self.draw_ellipse( pygame.Surface((3, 3)), (0, 10, 0, 50), rect, rect.w // 2 + 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) bounds_rect = self.draw_ellipse( pygame.Surface((3, 3)), (0, 10, 0, 50), rect, rect.h // 2 + 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__kwargs(self): """Ensures draw ellipse accepts the correct kwargs with and without a width arg. """ kwargs_list = [ { "surface": pygame.Surface((4, 4)), "color": pygame.Color("yellow"), "rect": pygame.Rect((0, 0), (3, 2)), "width": 1, }, { "surface": pygame.Surface((2, 1)), "color": (0, 10, 20), "rect": (0, 0, 1, 1), }, ] for kwargs in kwargs_list: bounds_rect = self.draw_ellipse(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__kwargs_order_independent(self): """Ensures draw ellipse's kwargs are not order dependent.""" bounds_rect = self.draw_ellipse( color=(1, 2, 3), surface=pygame.Surface((3, 2)), width=0, rect=pygame.Rect((1, 0), (1, 1)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__args_missing(self): """Ensures draw ellipse detects any missing required args.""" surface = pygame.Surface((1, 1)) with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(surface, pygame.Color("red")) with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse() def test_ellipse__kwargs_missing(self): """Ensures draw ellipse detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((1, 2)), "color": pygame.Color("red"), "rect": pygame.Rect((1, 0), (2, 2)), "width": 2, } for name in ("rect", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(**invalid_kwargs) def test_ellipse__arg_invalid_types(self): """Ensures draw ellipse detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") rect = pygame.Rect((1, 1), (1, 1)) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_ellipse(surface, color, rect, "1") with self.assertRaises(TypeError): # Invalid rect. bounds_rect = self.draw_ellipse(surface, color, (1, 2, 3, 4, 5), 1) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_ellipse(surface, 2.3, rect, 0) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_ellipse(rect, color, rect, 2) def test_ellipse__kwarg_invalid_types(self): """Ensures draw ellipse detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") rect = pygame.Rect((0, 1), (1, 1)) kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "rect": rect, "width": 1, }, { "surface": surface, "color": 2.3, # Invalid color. "rect": rect, "width": 1, }, { "surface": surface, "color": color, "rect": (0, 0, 0), # Invalid rect. "width": 1, }, {"surface": surface, "color": color, "rect": rect, "width": 1.1}, ] # Invalid width. for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(**kwargs) def test_ellipse__kwarg_invalid_name(self): """Ensures draw ellipse detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") rect = pygame.Rect((0, 1), (2, 2)) kwargs_list = [ { "surface": surface, "color": color, "rect": rect, "width": 1, "invalid": 1, }, {"surface": surface, "color": color, "rect": rect, "invalid": 1}, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(**kwargs) def test_ellipse__args_and_kwargs(self): """Ensures draw ellipse accepts a combination of args/kwargs""" surface = pygame.Surface((3, 1)) color = (255, 255, 0, 0) rect = pygame.Rect((1, 0), (2, 1)) width = 0 kwargs = {"surface": surface, "color": color, "rect": rect, "width": width} for name in ("surface", "color", "rect", "width"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_ellipse(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_ellipse(surface, color, **kwargs) elif "rect" == name: bounds_rect = self.draw_ellipse(surface, color, rect, **kwargs) else: bounds_rect = self.draw_ellipse(surface, color, rect, width, **kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__valid_width_values(self): """Ensures draw ellipse accepts different width values.""" pos = (1, 1) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) color = (10, 20, 30, 255) kwargs = { "surface": surface, "color": color, "rect": pygame.Rect(pos, (3, 2)), "width": None, } for width in (-1000, -10, -1, 0, 1, 10, 1000): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = color if width >= 0 else surface_color bounds_rect = self.draw_ellipse(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__valid_rect_formats(self): """Ensures draw ellipse accepts different rect formats.""" pos = (1, 1) expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = {"surface": surface, "color": expected_color, "rect": None, "width": 0} rects = (pygame.Rect(pos, (1, 3)), (pos, (2, 1)), (pos[0], pos[1], 1, 1)) for rect in rects: surface.fill(surface_color) # Clear for each test. kwargs["rect"] = rect bounds_rect = self.draw_ellipse(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__valid_color_formats(self): """Ensures draw ellipse accepts different color formats.""" pos = (1, 1) green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": None, "rect": pygame.Rect(pos, (1, 2)), "width": 0, } reds = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in reds: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_ellipse(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_ellipse__invalid_color_formats(self): """Ensures draw ellipse handles invalid color formats correctly.""" pos = (1, 1) surface = pygame.Surface((4, 3)) kwargs = { "surface": surface, "color": None, "rect": pygame.Rect(pos, (2, 2)), "width": 1, } for expected_color in (2.3, surface): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_ellipse(**kwargs) def test_ellipse(self): """Tests ellipses of differing sizes on surfaces of differing sizes. Checks if the number of sides touching the border of the surface is correct. """ left_top = [(0, 0), (1, 0), (0, 1), (1, 1)] sizes = [(4, 4), (5, 4), (4, 5), (5, 5)] color = (1, 13, 24, 255) def same_size(width, height, border_width): """Test for ellipses with the same size as the surface.""" surface = pygame.Surface((width, height)) self.draw_ellipse(surface, color, (0, 0, width, height), border_width) # For each of the four borders check if it contains the color borders = get_border_values(surface, width, height) for border in borders: self.assertTrue(color in border) def not_same_size(width, height, border_width, left, top): """Test for ellipses that aren't the same size as the surface.""" surface = pygame.Surface((width, height)) self.draw_ellipse( surface, color, (left, top, width - 1, height - 1), border_width ) borders = get_border_values(surface, width, height) # Check if two sides of the ellipse are touching the border sides_touching = [color in border for border in borders].count(True) self.assertEqual(sides_touching, 2) for width, height in sizes: for border_width in (0, 1): same_size(width, height, border_width) for left, top in left_top: not_same_size(width, height, border_width, left, top) def test_ellipse__big_ellipse(self): """Test for big ellipse that could overflow in algorithm""" width = 1025 height = 1025 border = 1 x_value_test = int(0.4 * height) y_value_test = int(0.4 * height) surface = pygame.Surface((width, height)) self.draw_ellipse(surface, (255, 0, 0), (0, 0, width, height), border) colored_pixels = 0 for y in range(height): if surface.get_at((x_value_test, y)) == (255, 0, 0): colored_pixels += 1 for x in range(width): if surface.get_at((x, y_value_test)) == (255, 0, 0): colored_pixels += 1 self.assertEqual(colored_pixels, border * 4) def test_ellipse__thick_line(self): """Ensures a thick lined ellipse is drawn correctly.""" ellipse_color = pygame.Color("yellow") surface_color = pygame.Color("black") surface = pygame.Surface((40, 40)) rect = pygame.Rect((0, 0), (31, 23)) rect.center = surface.get_rect().center # As the lines get thicker the internals of the ellipse are not # cleanly defined. So only test up to a few thicknesses before the # maximum thickness. for thickness in range(1, min(*rect.size) // 2 - 2): surface.fill(surface_color) # Clear for each test. self.draw_ellipse(surface, ellipse_color, rect, thickness) surface.lock() # For possible speed up. # Check vertical thickness on the ellipse's top. x = rect.centerx y_start = rect.top y_end = rect.top + thickness - 1 for y in range(y_start, y_end + 1): self.assertEqual(surface.get_at((x, y)), ellipse_color, thickness) # Check pixels above and below this line. self.assertEqual(surface.get_at((x, y_start - 1)), surface_color, thickness) self.assertEqual(surface.get_at((x, y_end + 1)), surface_color, thickness) # Check vertical thickness on the ellipse's bottom. x = rect.centerx y_start = rect.bottom - thickness y_end = rect.bottom - 1 for y in range(y_start, y_end + 1): self.assertEqual(surface.get_at((x, y)), ellipse_color, thickness) # Check pixels above and below this line. self.assertEqual(surface.get_at((x, y_start - 1)), surface_color, thickness) self.assertEqual(surface.get_at((x, y_end + 1)), surface_color, thickness) # Check horizontal thickness on the ellipse's left. x_start = rect.left x_end = rect.left + thickness - 1 y = rect.centery for x in range(x_start, x_end + 1): self.assertEqual(surface.get_at((x, y)), ellipse_color, thickness) # Check pixels to the left and right of this line. self.assertEqual(surface.get_at((x_start - 1, y)), surface_color, thickness) self.assertEqual(surface.get_at((x_end + 1, y)), surface_color, thickness) # Check horizontal thickness on the ellipse's right. x_start = rect.right - thickness x_end = rect.right - 1 y = rect.centery for x in range(x_start, x_end + 1): self.assertEqual(surface.get_at((x, y)), ellipse_color, thickness) # Check pixels to the left and right of this line. self.assertEqual(surface.get_at((x_start - 1, y)), surface_color, thickness) self.assertEqual(surface.get_at((x_end + 1, y)), surface_color, thickness) surface.unlock() def test_ellipse__no_holes(self): width = 80 height = 70 surface = pygame.Surface((width + 1, height)) rect = pygame.Rect(0, 0, width, height) for thickness in range(1, 37, 5): surface.fill("BLACK") self.draw_ellipse(surface, "RED", rect, thickness) for y in range(height): number_of_changes = 0 drawn_pixel = False for x in range(width + 1): if ( not drawn_pixel and surface.get_at((x, y)) == pygame.Color("RED") or drawn_pixel and surface.get_at((x, y)) == pygame.Color("BLACK") ): drawn_pixel = not drawn_pixel number_of_changes += 1 if y < thickness or y > height - thickness - 1: self.assertEqual(number_of_changes, 2) else: self.assertEqual(number_of_changes, 4) def test_ellipse__max_width(self): """Ensures an ellipse with max width (and greater) is drawn correctly.""" ellipse_color = pygame.Color("yellow") surface_color = pygame.Color("black") surface = pygame.Surface((40, 40)) rect = pygame.Rect((0, 0), (31, 21)) rect.center = surface.get_rect().center max_thickness = (min(*rect.size) + 1) // 2 for thickness in range(max_thickness, max_thickness + 3): surface.fill(surface_color) # Clear for each test. self.draw_ellipse(surface, ellipse_color, rect, thickness) surface.lock() # For possible speed up. # Check vertical thickness. for y in range(rect.top, rect.bottom): self.assertEqual(surface.get_at((rect.centerx, y)), ellipse_color) # Check horizontal thickness. for x in range(rect.left, rect.right): self.assertEqual(surface.get_at((x, rect.centery)), ellipse_color) # Check pixels above and below ellipse. self.assertEqual( surface.get_at((rect.centerx, rect.top - 1)), surface_color ) self.assertEqual( surface.get_at((rect.centerx, rect.bottom + 1)), surface_color ) # Check pixels to the left and right of the ellipse. self.assertEqual( surface.get_at((rect.left - 1, rect.centery)), surface_color ) self.assertEqual( surface.get_at((rect.right + 1, rect.centery)), surface_color ) surface.unlock() def _check_1_pixel_sized_ellipse( self, surface, collide_rect, surface_color, ellipse_color ): # Helper method to check the surface for 1 pixel wide and/or high # ellipses. surf_w, surf_h = surface.get_size() surface.lock() # For possible speed up. for pos in ((x, y) for y in range(surf_h) for x in range(surf_w)): # Since the ellipse is just a line we can use a rect to help find # where it is expected to be drawn. if collide_rect.collidepoint(pos): expected_color = ellipse_color else: expected_color = surface_color self.assertEqual( surface.get_at(pos), expected_color, f"collide_rect={collide_rect}, pos={pos}", ) surface.unlock() def test_ellipse__1_pixel_width(self): """Ensures an ellipse with a width of 1 is drawn correctly. An ellipse with a width of 1 pixel is a vertical line. """ ellipse_color = pygame.Color("red") surface_color = pygame.Color("black") surf_w, surf_h = 10, 20 surface = pygame.Surface((surf_w, surf_h)) rect = pygame.Rect((0, 0), (1, 0)) collide_rect = rect.copy() # Calculate some positions. off_left = -1 off_right = surf_w off_bottom = surf_h center_x = surf_w // 2 center_y = surf_h // 2 # Test some even and odd heights. for ellipse_h in range(6, 10): collide_rect.h = ellipse_h rect.h = ellipse_h # Calculate some variable positions. off_top = -(ellipse_h + 1) half_off_top = -(ellipse_h // 2) half_off_bottom = surf_h - (ellipse_h // 2) # Draw the ellipse in different positions: fully on-surface, # partially off-surface, and fully off-surface. positions = ( (off_left, off_top), (off_left, half_off_top), (off_left, center_y), (off_left, half_off_bottom), (off_left, off_bottom), (center_x, off_top), (center_x, half_off_top), (center_x, center_y), (center_x, half_off_bottom), (center_x, off_bottom), (off_right, off_top), (off_right, half_off_top), (off_right, center_y), (off_right, half_off_bottom), (off_right, off_bottom), ) for rect_pos in positions: surface.fill(surface_color) # Clear before each draw. rect.topleft = rect_pos collide_rect.topleft = rect_pos self.draw_ellipse(surface, ellipse_color, rect) self._check_1_pixel_sized_ellipse( surface, collide_rect, surface_color, ellipse_color ) def test_ellipse__1_pixel_width_spanning_surface(self): """Ensures an ellipse with a width of 1 is drawn correctly when spanning the height of the surface. An ellipse with a width of 1 pixel is a vertical line. """ ellipse_color = pygame.Color("red") surface_color = pygame.Color("black") surf_w, surf_h = 10, 20 surface = pygame.Surface((surf_w, surf_h)) rect = pygame.Rect((0, 0), (1, surf_h + 2)) # Longer than the surface. # Draw the ellipse in different positions: on-surface and off-surface. positions = ( (-1, -1), # (off_left, off_top) (0, -1), # (left_edge, off_top) (surf_w // 2, -1), # (center_x, off_top) (surf_w - 1, -1), # (right_edge, off_top) (surf_w, -1), ) # (off_right, off_top) for rect_pos in positions: surface.fill(surface_color) # Clear before each draw. rect.topleft = rect_pos self.draw_ellipse(surface, ellipse_color, rect) self._check_1_pixel_sized_ellipse( surface, rect, surface_color, ellipse_color ) def test_ellipse__1_pixel_height(self): """Ensures an ellipse with a height of 1 is drawn correctly. An ellipse with a height of 1 pixel is a horizontal line. """ ellipse_color = pygame.Color("red") surface_color = pygame.Color("black") surf_w, surf_h = 20, 10 surface = pygame.Surface((surf_w, surf_h)) rect = pygame.Rect((0, 0), (0, 1)) collide_rect = rect.copy() # Calculate some positions. off_right = surf_w off_top = -1 off_bottom = surf_h center_x = surf_w // 2 center_y = surf_h // 2 # Test some even and odd widths. for ellipse_w in range(6, 10): collide_rect.w = ellipse_w rect.w = ellipse_w # Calculate some variable positions. off_left = -(ellipse_w + 1) half_off_left = -(ellipse_w // 2) half_off_right = surf_w - (ellipse_w // 2) # Draw the ellipse in different positions: fully on-surface, # partially off-surface, and fully off-surface. positions = ( (off_left, off_top), (half_off_left, off_top), (center_x, off_top), (half_off_right, off_top), (off_right, off_top), (off_left, center_y), (half_off_left, center_y), (center_x, center_y), (half_off_right, center_y), (off_right, center_y), (off_left, off_bottom), (half_off_left, off_bottom), (center_x, off_bottom), (half_off_right, off_bottom), (off_right, off_bottom), ) for rect_pos in positions: surface.fill(surface_color) # Clear before each draw. rect.topleft = rect_pos collide_rect.topleft = rect_pos self.draw_ellipse(surface, ellipse_color, rect) self._check_1_pixel_sized_ellipse( surface, collide_rect, surface_color, ellipse_color ) def test_ellipse__1_pixel_height_spanning_surface(self): """Ensures an ellipse with a height of 1 is drawn correctly when spanning the width of the surface. An ellipse with a height of 1 pixel is a horizontal line. """ ellipse_color = pygame.Color("red") surface_color = pygame.Color("black") surf_w, surf_h = 20, 10 surface = pygame.Surface((surf_w, surf_h)) rect = pygame.Rect((0, 0), (surf_w + 2, 1)) # Wider than the surface. # Draw the ellipse in different positions: on-surface and off-surface. positions = ( (-1, -1), # (off_left, off_top) (-1, 0), # (off_left, top_edge) (-1, surf_h // 2), # (off_left, center_y) (-1, surf_h - 1), # (off_left, bottom_edge) (-1, surf_h), ) # (off_left, off_bottom) for rect_pos in positions: surface.fill(surface_color) # Clear before each draw. rect.topleft = rect_pos self.draw_ellipse(surface, ellipse_color, rect) self._check_1_pixel_sized_ellipse( surface, rect, surface_color, ellipse_color ) def test_ellipse__1_pixel_width_and_height(self): """Ensures an ellipse with a width and height of 1 is drawn correctly. An ellipse with a width and height of 1 pixel is a single pixel. """ ellipse_color = pygame.Color("red") surface_color = pygame.Color("black") surf_w, surf_h = 10, 10 surface = pygame.Surface((surf_w, surf_h)) rect = pygame.Rect((0, 0), (1, 1)) # Calculate some positions. off_left = -1 off_right = surf_w off_top = -1 off_bottom = surf_h left_edge = 0 right_edge = surf_w - 1 top_edge = 0 bottom_edge = surf_h - 1 center_x = surf_w // 2 center_y = surf_h // 2 # Draw the ellipse in different positions: center surface, # top/bottom/left/right edges, and off-surface. positions = ( (off_left, off_top), (off_left, top_edge), (off_left, center_y), (off_left, bottom_edge), (off_left, off_bottom), (left_edge, off_top), (left_edge, top_edge), (left_edge, center_y), (left_edge, bottom_edge), (left_edge, off_bottom), (center_x, off_top), (center_x, top_edge), (center_x, center_y), (center_x, bottom_edge), (center_x, off_bottom), (right_edge, off_top), (right_edge, top_edge), (right_edge, center_y), (right_edge, bottom_edge), (right_edge, off_bottom), (off_right, off_top), (off_right, top_edge), (off_right, center_y), (off_right, bottom_edge), (off_right, off_bottom), ) for rect_pos in positions: surface.fill(surface_color) # Clear before each draw. rect.topleft = rect_pos self.draw_ellipse(surface, ellipse_color, rect) self._check_1_pixel_sized_ellipse( surface, rect, surface_color, ellipse_color ) def test_ellipse__bounding_rect(self): """Ensures draw ellipse returns the correct bounding rect. Tests ellipses on and off the surface and a range of width/thickness values. """ ellipse_color = pygame.Color("red") surf_color = pygame.Color("black") min_width = min_height = 5 max_width = max_height = 7 sizes = ((min_width, min_height), (max_width, max_height)) surface = pygame.Surface((20, 20), 0, 32) surf_rect = surface.get_rect() # Make a rect that is bigger than the surface to help test drawing # ellipses off and partially off the surface. big_rect = surf_rect.inflate(min_width * 2 + 1, min_height * 2 + 1) for pos in rect_corners_mids_and_center( surf_rect ) + rect_corners_mids_and_center(big_rect): # Each of the ellipse's rect position attributes will be set to # the pos value. for attr in RECT_POSITION_ATTRIBUTES: # Test using different rect sizes and thickness values. for width, height in sizes: ellipse_rect = pygame.Rect((0, 0), (width, height)) setattr(ellipse_rect, attr, pos) for thickness in (0, 1, 2, 3, min(width, height)): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_ellipse( surface, ellipse_color, ellipse_rect, thickness ) # Calculating the expected_rect after the ellipse # is drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, ellipse_rect.topleft ) self.assertEqual(bounding_rect, expected_rect) def test_ellipse__surface_clip(self): """Ensures draw ellipse respects a surface's clip area. Tests drawing the ellipse filled and unfilled. """ surfw = surfh = 30 ellipse_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the ellipse's pos. for width in (0, 1): # Filled and unfilled. # Test centering the ellipse along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the ellipse without the # clip area set. pos_rect.center = center surface.set_clip(None) surface.fill(surface_color) self.draw_ellipse(surface, ellipse_color, pos_rect, width) expected_pts = get_color_points(surface, ellipse_color, clip_rect) # Clear the surface and set the clip area. Redraw the ellipse # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_ellipse(surface, ellipse_color, pos_rect, width) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the ellipse_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = ellipse_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() class DrawEllipseTest(DrawEllipseMixin, DrawTestCase): """Test draw module function ellipse. This class inherits the general tests from DrawEllipseMixin. It is also the class to add any draw.ellipse specific tests to. """ # Commented out to avoid cluttering the test output. Add back in if draw_py # ever properly supports drawing ellipses. # @unittest.skip('draw_py.draw_ellipse not supported yet') # class PythonDrawEllipseTest(DrawEllipseMixin, PythonDrawTestCase): # """Test draw_py module function draw_ellipse. # # This class inherits the general tests from DrawEllipseMixin. It is also # the class to add any draw_py.draw_ellipse specific tests to. # """ ### Line/Lines/AALine/AALines Testing ######################################### class BaseLineMixin: """Mixin base for drawing various lines. This class contains general helper methods and setup for testing the different types of lines. """ COLORS = ( (0, 0, 0), (255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (255, 0, 255), (0, 255, 255), (255, 255, 255), ) @staticmethod def _create_surfaces(): # Create some surfaces with different sizes, depths, and flags. surfaces = [] for size in ((49, 49), (50, 50)): for depth in (8, 16, 24, 32): for flags in (0, SRCALPHA): surface = pygame.display.set_mode(size, flags, depth) surfaces.append(surface) surfaces.append(surface.convert_alpha()) return surfaces @staticmethod def _rect_lines(rect): # Yields pairs of end points and their reverse (to test symmetry). # Uses a rect with the points radiating from its midleft. for pt in rect_corners_mids_and_center(rect): if pt in [rect.midleft, rect.center]: # Don't bother with these points. continue yield (rect.midleft, pt) yield (pt, rect.midleft) ### Line Testing ############################################################## class LineMixin(BaseLineMixin): """Mixin test for drawing a single line. This class contains all the general single line drawing tests. """ def test_line__args(self): """Ensures draw line accepts the correct args.""" bounds_rect = self.draw_line( pygame.Surface((3, 3)), (0, 10, 0, 50), (0, 0), (1, 1), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__args_without_width(self): """Ensures draw line accepts the args without a width.""" bounds_rect = self.draw_line( pygame.Surface((2, 2)), (0, 0, 0, 50), (0, 0), (2, 2) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__kwargs(self): """Ensures draw line accepts the correct kwargs with and without a width arg. """ surface = pygame.Surface((4, 4)) color = pygame.Color("yellow") start_pos = (1, 1) end_pos = (2, 2) kwargs_list = [ { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "width": 1, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, }, ] for kwargs in kwargs_list: bounds_rect = self.draw_line(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__kwargs_order_independent(self): """Ensures draw line's kwargs are not order dependent.""" bounds_rect = self.draw_line( start_pos=(1, 2), end_pos=(2, 1), width=2, color=(10, 20, 30), surface=pygame.Surface((3, 2)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__args_missing(self): """Ensures draw line detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_line(surface, color, (0, 0)) with self.assertRaises(TypeError): bounds_rect = self.draw_line(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_line(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_line() def test_line__kwargs_missing(self): """Ensures draw line detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((3, 2)), "color": pygame.Color("red"), "start_pos": (2, 1), "end_pos": (2, 2), "width": 1, } for name in ("end_pos", "start_pos", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_line(**invalid_kwargs) def test_line__arg_invalid_types(self): """Ensures draw line detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") start_pos = (0, 1) end_pos = (1, 2) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_line(surface, color, start_pos, end_pos, "1") with self.assertRaises(TypeError): # Invalid end_pos. bounds_rect = self.draw_line(surface, color, start_pos, (1, 2, 3)) with self.assertRaises(TypeError): # Invalid start_pos. bounds_rect = self.draw_line(surface, color, (1,), end_pos) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_line(surface, 2.3, start_pos, end_pos) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_line((1, 2, 3, 4), color, start_pos, end_pos) def test_line__kwarg_invalid_types(self): """Ensures draw line detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") start_pos = (1, 0) end_pos = (2, 0) width = 1 kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "start_pos": start_pos, "end_pos": end_pos, "width": width, }, { "surface": surface, "color": 2.3, # Invalid color. "start_pos": start_pos, "end_pos": end_pos, "width": width, }, { "surface": surface, "color": color, "start_pos": (0, 0, 0), # Invalid start_pos. "end_pos": end_pos, "width": width, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": (0,), # Invalid end_pos. "width": width, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "width": 1.2, }, ] # Invalid width. for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_line(**kwargs) def test_line__kwarg_invalid_name(self): """Ensures draw line detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") start_pos = (1, 1) end_pos = (2, 0) kwargs_list = [ { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "width": 1, "invalid": 1, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_line(**kwargs) def test_line__args_and_kwargs(self): """Ensures draw line accepts a combination of args/kwargs""" surface = pygame.Surface((3, 2)) color = (255, 255, 0, 0) start_pos = (0, 1) end_pos = (1, 2) width = 0 kwargs = { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "width": width, } for name in ("surface", "color", "start_pos", "end_pos", "width"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_line(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_line(surface, color, **kwargs) elif "start_pos" == name: bounds_rect = self.draw_line(surface, color, start_pos, **kwargs) elif "end_pos" == name: bounds_rect = self.draw_line( surface, color, start_pos, end_pos, **kwargs ) else: bounds_rect = self.draw_line( surface, color, start_pos, end_pos, width, **kwargs ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__valid_width_values(self): """Ensures draw line accepts different width values.""" line_color = pygame.Color("yellow") surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) pos = (2, 1) kwargs = { "surface": surface, "color": line_color, "start_pos": pos, "end_pos": (2, 2), "width": None, } for width in (-100, -10, -1, 0, 1, 10, 100): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = line_color if width > 0 else surface_color bounds_rect = self.draw_line(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__valid_start_pos_formats(self): """Ensures draw line accepts different start_pos formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = { "surface": surface, "color": expected_color, "start_pos": None, "end_pos": (2, 2), "width": 2, } x, y = 2, 1 # start position # The point values can be ints or floats. for start_pos in ((x, y), (x + 0.1, y), (x, y + 0.1), (x + 0.1, y + 0.1)): # The point type can be a tuple/list/Vector2. for seq_type in (tuple, list, Vector2): surface.fill(surface_color) # Clear for each test. kwargs["start_pos"] = seq_type(start_pos) bounds_rect = self.draw_line(**kwargs) self.assertEqual(surface.get_at((x, y)), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__valid_end_pos_formats(self): """Ensures draw line accepts different end_pos formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = { "surface": surface, "color": expected_color, "start_pos": (2, 1), "end_pos": None, "width": 2, } x, y = 2, 2 # end position # The point values can be ints or floats. for end_pos in ((x, y), (x + 0.2, y), (x, y + 0.2), (x + 0.2, y + 0.2)): # The point type can be a tuple/list/Vector2. for seq_type in (tuple, list, Vector2): surface.fill(surface_color) # Clear for each test. kwargs["end_pos"] = seq_type(end_pos) bounds_rect = self.draw_line(**kwargs) self.assertEqual(surface.get_at((x, y)), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__invalid_start_pos_formats(self): """Ensures draw line handles invalid start_pos formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "start_pos": None, "end_pos": (2, 2), "width": 1, } start_pos_fmts = ( (2,), # Too few coords. (2, 1, 0), # Too many coords. (2, "1"), # Wrong type. {2, 1}, # Wrong type. dict(((2, 1),)), ) # Wrong type. for start_pos in start_pos_fmts: kwargs["start_pos"] = start_pos with self.assertRaises(TypeError): bounds_rect = self.draw_line(**kwargs) def test_line__invalid_end_pos_formats(self): """Ensures draw line handles invalid end_pos formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "start_pos": (2, 2), "end_pos": None, "width": 1, } end_pos_fmts = ( (2,), # Too few coords. (2, 1, 0), # Too many coords. (2, "1"), # Wrong type. {2, 1}, # Wrong type. dict(((2, 1),)), ) # Wrong type. for end_pos in end_pos_fmts: kwargs["end_pos"] = end_pos with self.assertRaises(TypeError): bounds_rect = self.draw_line(**kwargs) def test_line__valid_color_formats(self): """Ensures draw line accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) pos = (1, 1) kwargs = { "surface": surface, "color": None, "start_pos": pos, "end_pos": (2, 1), "width": 3, } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_line(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_line__invalid_color_formats(self): """Ensures draw line handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "start_pos": (1, 1), "end_pos": (2, 1), "width": 1, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_line(**kwargs) def test_line__color(self): """Tests if the line drawn is the correct color.""" pos = (0, 0) for surface in self._create_surfaces(): for expected_color in self.COLORS: self.draw_line(surface, expected_color, pos, (1, 0)) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_line__color_with_thickness(self): """Ensures a thick line is drawn using the correct color.""" from_x = 5 to_x = 10 y = 5 for surface in self._create_surfaces(): for expected_color in self.COLORS: self.draw_line(surface, expected_color, (from_x, y), (to_x, y), 5) for pos in ((x, y + i) for i in (-2, 0, 2) for x in (from_x, to_x)): self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_line__gaps(self): """Tests if the line drawn contains any gaps.""" expected_color = (255, 255, 255) for surface in self._create_surfaces(): width = surface.get_width() self.draw_line(surface, expected_color, (0, 0), (width - 1, 0)) for x in range(width): pos = (x, 0) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_line__gaps_with_thickness(self): """Ensures a thick line is drawn without any gaps.""" expected_color = (255, 255, 255) thickness = 5 for surface in self._create_surfaces(): width = surface.get_width() - 1 h = width // 5 w = h * 5 self.draw_line(surface, expected_color, (0, 5), (w, 5 + h), thickness) for x in range(w + 1): for y in range(3, 8): pos = (x, y + ((x + 2) // 5)) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_line__bounding_rect(self): """Ensures draw line returns the correct bounding rect. Tests lines with endpoints on and off the surface and a range of width/thickness values. """ if isinstance(self, PythonDrawTestCase): self.skipTest("bounding rects not supported in draw_py.draw_line") line_color = pygame.Color("red") surf_color = pygame.Color("black") width = height = 30 # Using a rect to help manage where the lines are drawn. helper_rect = pygame.Rect((0, 0), (width, height)) # Testing surfaces of different sizes. One larger than the helper_rect # and one smaller (to test lines that span the surface). for size in ((width + 5, height + 5), (width - 5, height - 5)): surface = pygame.Surface(size, 0, 32) surf_rect = surface.get_rect() # Move the helper rect to different positions to test line # endpoints on and off the surface. for pos in rect_corners_mids_and_center(surf_rect): helper_rect.center = pos # Draw using different thicknesses. for thickness in range(-1, 5): for start, end in self._rect_lines(helper_rect): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_line( surface, line_color, start, end, thickness ) if 0 < thickness: # Calculating the expected_rect after the line is # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, start ) else: # Nothing drawn. expected_rect = pygame.Rect(start, (0, 0)) self.assertEqual( bounding_rect, expected_rect, "start={}, end={}, size={}, thickness={}".format( start, end, size, thickness ), ) def test_line__surface_clip(self): """Ensures draw line respects a surface's clip area.""" surfw = surfh = 30 line_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the line's pos. for thickness in (1, 3): # Test different line widths. # Test centering the line along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the line without the # clip area set. pos_rect.center = center surface.set_clip(None) surface.fill(surface_color) self.draw_line( surface, line_color, pos_rect.midtop, pos_rect.midbottom, thickness ) expected_pts = get_color_points(surface, line_color, clip_rect) # Clear the surface and set the clip area. Redraw the line # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_line( surface, line_color, pos_rect.midtop, pos_rect.midbottom, thickness ) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the line_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = line_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() # Commented out to avoid cluttering the test output. Add back in if draw_py # ever fully supports drawing single lines. # @unittest.skip('draw_py.draw_line not fully supported yet') # class PythonDrawLineTest(LineMixin, PythonDrawTestCase): # """Test draw_py module function line. # # This class inherits the general tests from LineMixin. It is also the class # to add any draw_py.draw_line specific tests to. # """ class DrawLineTest(LineMixin, DrawTestCase): """Test draw module function line. This class inherits the general tests from LineMixin. It is also the class to add any draw.line specific tests to. """ def test_line_endianness(self): """test color component order""" for depth in (24, 32): surface = pygame.Surface((5, 3), 0, depth) surface.fill(pygame.Color(0, 0, 0)) self.draw_line(surface, pygame.Color(255, 0, 0), (0, 1), (2, 1), 1) self.assertGreater(surface.get_at((1, 1)).r, 0, "there should be red here") surface.fill(pygame.Color(0, 0, 0)) self.draw_line(surface, pygame.Color(0, 0, 255), (0, 1), (2, 1), 1) self.assertGreater(surface.get_at((1, 1)).b, 0, "there should be blue here") def test_line(self): # (l, t), (l, t) self.surf_size = (320, 200) self.surf = pygame.Surface(self.surf_size, pygame.SRCALPHA) self.color = (1, 13, 24, 205) drawn = draw.line(self.surf, self.color, (1, 0), (200, 0)) self.assertEqual( drawn.right, 201, "end point arg should be (or at least was) inclusive" ) # Should be colored where it's supposed to be for pt in test_utils.rect_area_pts(drawn): self.assertEqual(self.surf.get_at(pt), self.color) # And not where it shouldn't for pt in test_utils.rect_outer_bounds(drawn): self.assertNotEqual(self.surf.get_at(pt), self.color) # Line width greater that 1 line_width = 2 offset = 5 a = (offset, offset) b = (self.surf_size[0] - offset, a[1]) c = (a[0], self.surf_size[1] - offset) d = (b[0], c[1]) e = (a[0] + offset, c[1]) f = (b[0], c[0] + 5) lines = [ (a, d), (b, c), (c, b), (d, a), (a, b), (b, a), (a, c), (c, a), (a, e), (e, a), (a, f), (f, a), (a, a), ] for p1, p2 in lines: msg = f"{p1} - {p2}" if p1[0] <= p2[0]: plow = p1 phigh = p2 else: plow = p2 phigh = p1 self.surf.fill((0, 0, 0)) rec = draw.line(self.surf, (255, 255, 255), p1, p2, line_width) xinc = yinc = 0 if abs(p1[0] - p2[0]) > abs(p1[1] - p2[1]): yinc = 1 else: xinc = 1 for i in range(line_width): p = (p1[0] + xinc * i, p1[1] + yinc * i) self.assertEqual(self.surf.get_at(p), (255, 255, 255), msg) p = (p2[0] + xinc * i, p2[1] + yinc * i) self.assertEqual(self.surf.get_at(p), (255, 255, 255), msg) p = (plow[0] - 1, plow[1]) self.assertEqual(self.surf.get_at(p), (0, 0, 0), msg) p = (plow[0] + xinc * line_width, plow[1] + yinc * line_width) self.assertEqual(self.surf.get_at(p), (0, 0, 0), msg) p = (phigh[0] + xinc * line_width, phigh[1] + yinc * line_width) self.assertEqual(self.surf.get_at(p), (0, 0, 0), msg) if p1[0] < p2[0]: rx = p1[0] else: rx = p2[0] if p1[1] < p2[1]: ry = p1[1] else: ry = p2[1] w = abs(p2[0] - p1[0]) + 1 + xinc * (line_width - 1) h = abs(p2[1] - p1[1]) + 1 + yinc * (line_width - 1) msg += f", {rec}" self.assertEqual(rec, (rx, ry, w, h), msg) def test_line_for_gaps(self): # This checks bug Thick Line Bug #448 width = 200 height = 200 surf = pygame.Surface((width, height), pygame.SRCALPHA) def white_surrounded_pixels(x, y): offsets = [(1, 0), (0, 1), (-1, 0), (0, -1)] WHITE = (255, 255, 255, 255) return len( [1 for dx, dy in offsets if surf.get_at((x + dx, y + dy)) == WHITE] ) def check_white_line(start, end): surf.fill((0, 0, 0)) pygame.draw.line(surf, (255, 255, 255), start, end, 30) BLACK = (0, 0, 0, 255) for x in range(1, width - 1): for y in range(1, height - 1): if surf.get_at((x, y)) == BLACK: self.assertTrue(white_surrounded_pixels(x, y) < 3) check_white_line((50, 50), (140, 0)) check_white_line((50, 50), (0, 120)) check_white_line((50, 50), (199, 198)) ### Lines Testing ############################################################# class LinesMixin(BaseLineMixin): """Mixin test for drawing lines. This class contains all the general lines drawing tests. """ def test_lines__args(self): """Ensures draw lines accepts the correct args.""" bounds_rect = self.draw_lines( pygame.Surface((3, 3)), (0, 10, 0, 50), False, ((0, 0), (1, 1)), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__args_without_width(self): """Ensures draw lines accepts the args without a width.""" bounds_rect = self.draw_lines( pygame.Surface((2, 2)), (0, 0, 0, 50), False, ((0, 0), (1, 1)) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__kwargs(self): """Ensures draw lines accepts the correct kwargs with and without a width arg. """ surface = pygame.Surface((4, 4)) color = pygame.Color("yellow") points = ((0, 0), (1, 1), (2, 2)) kwargs_list = [ { "surface": surface, "color": color, "closed": False, "points": points, "width": 1, }, {"surface": surface, "color": color, "closed": False, "points": points}, ] for kwargs in kwargs_list: bounds_rect = self.draw_lines(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__kwargs_order_independent(self): """Ensures draw lines's kwargs are not order dependent.""" bounds_rect = self.draw_lines( closed=1, points=((0, 0), (1, 1), (2, 2)), width=2, color=(10, 20, 30), surface=pygame.Surface((3, 2)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__args_missing(self): """Ensures draw lines detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_lines(surface, color, 0) with self.assertRaises(TypeError): bounds_rect = self.draw_lines(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_lines(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_lines() def test_lines__kwargs_missing(self): """Ensures draw lines detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((3, 2)), "color": pygame.Color("red"), "closed": 1, "points": ((2, 2), (1, 1)), "width": 1, } for name in ("points", "closed", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_lines(**invalid_kwargs) def test_lines__arg_invalid_types(self): """Ensures draw lines detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") closed = 0 points = ((1, 2), (2, 1)) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_lines(surface, color, closed, points, "1") with self.assertRaises(TypeError): # Invalid points. bounds_rect = self.draw_lines(surface, color, closed, (1, 2, 3)) with self.assertRaises(TypeError): # Invalid closed. bounds_rect = self.draw_lines(surface, color, InvalidBool(), points) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_lines(surface, 2.3, closed, points) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_lines((1, 2, 3, 4), color, closed, points) def test_lines__kwarg_invalid_types(self): """Ensures draw lines detects invalid kwarg types.""" valid_kwargs = { "surface": pygame.Surface((3, 3)), "color": pygame.Color("green"), "closed": False, "points": ((1, 2), (2, 1)), "width": 1, } invalid_kwargs = { "surface": pygame.Surface, "color": 2.3, "closed": InvalidBool(), "points": (0, 0, 0), "width": 1.2, } for kwarg in ("surface", "color", "closed", "points", "width"): kwargs = dict(valid_kwargs) kwargs[kwarg] = invalid_kwargs[kwarg] with self.assertRaises(TypeError): bounds_rect = self.draw_lines(**kwargs) def test_lines__kwarg_invalid_name(self): """Ensures draw lines detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") closed = 1 points = ((1, 2), (2, 1)) kwargs_list = [ { "surface": surface, "color": color, "closed": closed, "points": points, "width": 1, "invalid": 1, }, { "surface": surface, "color": color, "closed": closed, "points": points, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_lines(**kwargs) def test_lines__args_and_kwargs(self): """Ensures draw lines accepts a combination of args/kwargs""" surface = pygame.Surface((3, 2)) color = (255, 255, 0, 0) closed = 0 points = ((1, 2), (2, 1)) width = 1 kwargs = { "surface": surface, "color": color, "closed": closed, "points": points, "width": width, } for name in ("surface", "color", "closed", "points", "width"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_lines(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_lines(surface, color, **kwargs) elif "closed" == name: bounds_rect = self.draw_lines(surface, color, closed, **kwargs) elif "points" == name: bounds_rect = self.draw_lines(surface, color, closed, points, **kwargs) else: bounds_rect = self.draw_lines( surface, color, closed, points, width, **kwargs ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__valid_width_values(self): """Ensures draw lines accepts different width values.""" line_color = pygame.Color("yellow") surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) pos = (1, 1) kwargs = { "surface": surface, "color": line_color, "closed": False, "points": (pos, (2, 1)), "width": None, } for width in (-100, -10, -1, 0, 1, 10, 100): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = line_color if width > 0 else surface_color bounds_rect = self.draw_lines(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__valid_points_format(self): """Ensures draw lines accepts different points formats.""" expected_color = (10, 20, 30, 255) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": expected_color, "closed": False, "points": None, "width": 1, } # The point type can be a tuple/list/Vector2. point_types = ( (tuple, tuple, tuple, tuple), # all tuples (list, list, list, list), # all lists (Vector2, Vector2, Vector2, Vector2), # all Vector2s (list, Vector2, tuple, Vector2), ) # mix # The point values can be ints or floats. point_values = ( ((1, 1), (2, 1), (2, 2), (1, 2)), ((1, 1), (2.2, 1), (2.1, 2.2), (1, 2.1)), ) # Each sequence of points can be a tuple or a list. seq_types = (tuple, list) for point_type in point_types: for values in point_values: check_pos = values[0] points = [point_type[i](pt) for i, pt in enumerate(values)] for seq_type in seq_types: surface.fill(surface_color) # Clear for each test. kwargs["points"] = seq_type(points) bounds_rect = self.draw_lines(**kwargs) self.assertEqual(surface.get_at(check_pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__invalid_points_formats(self): """Ensures draw lines handles invalid points formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "closed": False, "points": None, "width": 1, } points_fmts = ( ((1, 1), (2,)), # Too few coords. ((1, 1), (2, 2, 2)), # Too many coords. ((1, 1), (2, "2")), # Wrong type. ((1, 1), {2, 3}), # Wrong type. ((1, 1), dict(((2, 2), (3, 3)))), # Wrong type. {(1, 1), (1, 2)}, # Wrong type. dict(((1, 1), (4, 4))), ) # Wrong type. for points in points_fmts: kwargs["points"] = points with self.assertRaises(TypeError): bounds_rect = self.draw_lines(**kwargs) def test_lines__invalid_points_values(self): """Ensures draw lines handles invalid points values correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "closed": False, "points": None, "width": 1, } for points in ([], ((1, 1),)): # Too few points. for seq_type in (tuple, list): # Test as tuples and lists. kwargs["points"] = seq_type(points) with self.assertRaises(ValueError): bounds_rect = self.draw_lines(**kwargs) def test_lines__valid_closed_values(self): """Ensures draw lines accepts different closed values.""" line_color = pygame.Color("blue") surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) pos = (1, 2) kwargs = { "surface": surface, "color": line_color, "closed": None, "points": ((1, 1), (3, 1), (3, 3), (1, 3)), "width": 1, } true_values = (-7, 1, 10, "2", 3.1, (4,), [5], True) false_values = (None, "", 0, (), [], False) for closed in true_values + false_values: surface.fill(surface_color) # Clear for each test. kwargs["closed"] = closed expected_color = line_color if closed else surface_color bounds_rect = self.draw_lines(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__valid_color_formats(self): """Ensures draw lines accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) pos = (1, 1) kwargs = { "surface": surface, "color": None, "closed": False, "points": (pos, (2, 1)), "width": 3, } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_lines(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_lines__invalid_color_formats(self): """Ensures draw lines handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "closed": False, "points": ((1, 1), (1, 2)), "width": 1, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_lines(**kwargs) def test_lines__color(self): """Tests if the lines drawn are the correct color. Draws lines around the border of the given surface and checks if all borders of the surface only contain the given color. """ for surface in self._create_surfaces(): for expected_color in self.COLORS: self.draw_lines(surface, expected_color, True, corners(surface)) for pos, color in border_pos_and_color(surface): self.assertEqual(color, expected_color, f"pos={pos}") def test_lines__color_with_thickness(self): """Ensures thick lines are drawn using the correct color.""" x_left = y_top = 5 for surface in self._create_surfaces(): x_right = surface.get_width() - 5 y_bottom = surface.get_height() - 5 endpoints = ( (x_left, y_top), (x_right, y_top), (x_right, y_bottom), (x_left, y_bottom), ) for expected_color in self.COLORS: self.draw_lines(surface, expected_color, True, endpoints, 3) for t in (-1, 0, 1): for x in range(x_left, x_right + 1): for y in (y_top, y_bottom): pos = (x, y + t) self.assertEqual( surface.get_at(pos), expected_color, f"pos={pos}", ) for y in range(y_top, y_bottom + 1): for x in (x_left, x_right): pos = (x + t, y) self.assertEqual( surface.get_at(pos), expected_color, f"pos={pos}", ) def test_lines__gaps(self): """Tests if the lines drawn contain any gaps. Draws lines around the border of the given surface and checks if all borders of the surface contain any gaps. """ expected_color = (255, 255, 255) for surface in self._create_surfaces(): self.draw_lines(surface, expected_color, True, corners(surface)) for pos, color in border_pos_and_color(surface): self.assertEqual(color, expected_color, f"pos={pos}") def test_lines__gaps_with_thickness(self): """Ensures thick lines are drawn without any gaps.""" expected_color = (255, 255, 255) x_left = y_top = 5 for surface in self._create_surfaces(): h = (surface.get_width() - 11) // 5 w = h * 5 x_right = x_left + w y_bottom = y_top + h endpoints = ((x_left, y_top), (x_right, y_top), (x_right, y_bottom)) self.draw_lines(surface, expected_color, True, endpoints, 3) for x in range(x_left, x_right + 1): for t in (-1, 0, 1): pos = (x, y_top + t) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") pos = (x, y_top + t + ((x - 3) // 5)) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") for y in range(y_top, y_bottom + 1): for t in (-1, 0, 1): pos = (x_right + t, y) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_lines__bounding_rect(self): """Ensures draw lines returns the correct bounding rect. Tests lines with endpoints on and off the surface and a range of width/thickness values. """ line_color = pygame.Color("red") surf_color = pygame.Color("black") width = height = 30 # Using a rect to help manage where the lines are drawn. pos_rect = pygame.Rect((0, 0), (width, height)) # Testing surfaces of different sizes. One larger than the pos_rect # and one smaller (to test lines that span the surface). for size in ((width + 5, height + 5), (width - 5, height - 5)): surface = pygame.Surface(size, 0, 32) surf_rect = surface.get_rect() # Move pos_rect to different positions to test line endpoints on # and off the surface. for pos in rect_corners_mids_and_center(surf_rect): pos_rect.center = pos # Shape: Triangle (if closed), ^ caret (if not closed). pts = (pos_rect.midleft, pos_rect.midtop, pos_rect.midright) pos = pts[0] # Rect position if nothing drawn. # Draw using different thickness and closed values. for thickness in range(-1, 5): for closed in (True, False): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_lines( surface, line_color, closed, pts, thickness ) if 0 < thickness: # Calculating the expected_rect after the lines are # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, pos ) else: # Nothing drawn. expected_rect = pygame.Rect(pos, (0, 0)) self.assertEqual(bounding_rect, expected_rect) def test_lines__surface_clip(self): """Ensures draw lines respects a surface's clip area.""" surfw = surfh = 30 line_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the lines's pos. # Test centering the pos_rect along the clip rect's edge to allow for # drawing the lines over the clip_rect's bounds. for center in rect_corners_mids_and_center(clip_rect): pos_rect.center = center pts = (pos_rect.midtop, pos_rect.center, pos_rect.midbottom) for closed in (True, False): # Test closed and not closed. for thickness in (1, 3): # Test different line widths. # Get the expected points by drawing the lines without the # clip area set. surface.set_clip(None) surface.fill(surface_color) self.draw_lines(surface, line_color, closed, pts, thickness) expected_pts = get_color_points(surface, line_color, clip_rect) # Clear the surface and set the clip area. Redraw the lines # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_lines(surface, line_color, closed, pts, thickness) surface.lock() # For possible speed up. # Check all the surface points to ensure only the # expected_pts are the line_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = line_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() # Commented out to avoid cluttering the test output. Add back in if draw_py # ever fully supports drawing lines. # class PythonDrawLinesTest(LinesMixin, PythonDrawTestCase): # """Test draw_py module function lines. # # This class inherits the general tests from LinesMixin. It is also the # class to add any draw_py.draw_lines specific tests to. # """ class DrawLinesTest(LinesMixin, DrawTestCase): """Test draw module function lines. This class inherits the general tests from LinesMixin. It is also the class to add any draw.lines specific tests to. """ ### AALine Testing ############################################################ class AALineMixin(BaseLineMixin): """Mixin test for drawing a single aaline. This class contains all the general single aaline drawing tests. """ def test_aaline__args(self): """Ensures draw aaline accepts the correct args.""" bounds_rect = self.draw_aaline( pygame.Surface((3, 3)), (0, 10, 0, 50), (0, 0), (1, 1), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__args_without_blend(self): """Ensures draw aaline accepts the args without a blend.""" bounds_rect = self.draw_aaline( pygame.Surface((2, 2)), (0, 0, 0, 50), (0, 0), (2, 2) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__blend_warning(self): """From pygame 2, blend=False should raise DeprecationWarning.""" with warnings.catch_warnings(record=True) as w: # Cause all warnings to always be triggered. warnings.simplefilter("always") # Trigger DeprecationWarning. self.draw_aaline( pygame.Surface((2, 2)), (0, 0, 0, 50), (0, 0), (2, 2), False ) # Check if there is only one warning and is a DeprecationWarning. self.assertEqual(len(w), 1) self.assertTrue(issubclass(w[-1].category, DeprecationWarning)) def test_aaline__kwargs(self): """Ensures draw aaline accepts the correct kwargs""" surface = pygame.Surface((4, 4)) color = pygame.Color("yellow") start_pos = (1, 1) end_pos = (2, 2) kwargs_list = [ { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, }, ] for kwargs in kwargs_list: bounds_rect = self.draw_aaline(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__kwargs_order_independent(self): """Ensures draw aaline's kwargs are not order dependent.""" bounds_rect = self.draw_aaline( start_pos=(1, 2), end_pos=(2, 1), color=(10, 20, 30), surface=pygame.Surface((3, 2)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__args_missing(self): """Ensures draw aaline detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(surface, color, (0, 0)) with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_aaline() def test_aaline__kwargs_missing(self): """Ensures draw aaline detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((3, 2)), "color": pygame.Color("red"), "start_pos": (2, 1), "end_pos": (2, 2), } for name in ("end_pos", "start_pos", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**invalid_kwargs) def test_aaline__arg_invalid_types(self): """Ensures draw aaline detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") start_pos = (0, 1) end_pos = (1, 2) with self.assertRaises(TypeError): # Invalid end_pos. bounds_rect = self.draw_aaline(surface, color, start_pos, (1, 2, 3)) with self.assertRaises(TypeError): # Invalid start_pos. bounds_rect = self.draw_aaline(surface, color, (1,), end_pos) with self.assertRaises(ValueError): # Invalid color. bounds_rect = self.draw_aaline(surface, "invalid-color", start_pos, end_pos) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_aaline((1, 2, 3, 4), color, start_pos, end_pos) def test_aaline__kwarg_invalid_types(self): """Ensures draw aaline detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") start_pos = (1, 0) end_pos = (2, 0) kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "start_pos": start_pos, "end_pos": end_pos, }, { "surface": surface, "color": 2.3, # Invalid color. "start_pos": start_pos, "end_pos": end_pos, }, { "surface": surface, "color": color, "start_pos": (0, 0, 0), # Invalid start_pos. "end_pos": end_pos, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": (0,), # Invalid end_pos. }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**kwargs) def test_aaline__kwarg_invalid_name(self): """Ensures draw aaline detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") start_pos = (1, 1) end_pos = (2, 0) kwargs_list = [ { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "invalid": 1, }, { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**kwargs) def test_aaline__args_and_kwargs(self): """Ensures draw aaline accepts a combination of args/kwargs""" surface = pygame.Surface((3, 2)) color = (255, 255, 0, 0) start_pos = (0, 1) end_pos = (1, 2) kwargs = { "surface": surface, "color": color, "start_pos": start_pos, "end_pos": end_pos, } for name in ("surface", "color", "start_pos", "end_pos"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_aaline(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_aaline(surface, color, **kwargs) elif "start_pos" == name: bounds_rect = self.draw_aaline(surface, color, start_pos, **kwargs) elif "end_pos" == name: bounds_rect = self.draw_aaline( surface, color, start_pos, end_pos, **kwargs ) else: bounds_rect = self.draw_aaline( surface, color, start_pos, end_pos, **kwargs ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__valid_start_pos_formats(self): """Ensures draw aaline accepts different start_pos formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = { "surface": surface, "color": expected_color, "start_pos": None, "end_pos": (2, 2), } x, y = 2, 1 # start position positions = ((x, y), (x + 0.01, y), (x, y + 0.01), (x + 0.01, y + 0.01)) for start_pos in positions: for seq_type in (tuple, list, Vector2): surface.fill(surface_color) # Clear for each test. kwargs["start_pos"] = seq_type(start_pos) bounds_rect = self.draw_aaline(**kwargs) color = surface.get_at((x, y)) for i, sub_color in enumerate(expected_color): # The color could be slightly off the expected color due to # any fractional position arguments. self.assertGreaterEqual(color[i] + 6, sub_color, start_pos) self.assertIsInstance(bounds_rect, pygame.Rect, start_pos) def test_aaline__valid_end_pos_formats(self): """Ensures draw aaline accepts different end_pos formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = { "surface": surface, "color": expected_color, "start_pos": (2, 1), "end_pos": None, } x, y = 2, 2 # end position positions = ((x, y), (x + 0.02, y), (x, y + 0.02), (x + 0.02, y + 0.02)) for end_pos in positions: for seq_type in (tuple, list, Vector2): surface.fill(surface_color) # Clear for each test. kwargs["end_pos"] = seq_type(end_pos) bounds_rect = self.draw_aaline(**kwargs) color = surface.get_at((x, y)) for i, sub_color in enumerate(expected_color): # The color could be slightly off the expected color due to # any fractional position arguments. self.assertGreaterEqual(color[i] + 15, sub_color, end_pos) self.assertIsInstance(bounds_rect, pygame.Rect, end_pos) def test_aaline__invalid_start_pos_formats(self): """Ensures draw aaline handles invalid start_pos formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "start_pos": None, "end_pos": (2, 2), } start_pos_fmts = ( (2,), # Too few coords. (2, 1, 0), # Too many coords. (2, "1"), # Wrong type. {2, 1}, # Wrong type. dict(((2, 1),)), ) # Wrong type. for start_pos in start_pos_fmts: kwargs["start_pos"] = start_pos with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**kwargs) def test_aaline__invalid_end_pos_formats(self): """Ensures draw aaline handles invalid end_pos formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "start_pos": (2, 2), "end_pos": None, } end_pos_fmts = ( (2,), # Too few coords. (2, 1, 0), # Too many coords. (2, "1"), # Wrong type. {2, 1}, # Wrong type. dict(((2, 1),)), ) # Wrong type. for end_pos in end_pos_fmts: kwargs["end_pos"] = end_pos with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**kwargs) def test_aaline__valid_color_formats(self): """Ensures draw aaline accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) pos = (1, 1) kwargs = { "surface": surface, "color": None, "start_pos": pos, "end_pos": (2, 1), } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_aaline(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aaline__invalid_color_formats(self): """Ensures draw aaline handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "start_pos": (1, 1), "end_pos": (2, 1), } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_aaline(**kwargs) def test_aaline__color(self): """Tests if the aaline drawn is the correct color.""" pos = (0, 0) for surface in self._create_surfaces(): for expected_color in self.COLORS: self.draw_aaline(surface, expected_color, pos, (1, 0)) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_aaline__gaps(self): """Tests if the aaline drawn contains any gaps. See: #512 """ expected_color = (255, 255, 255) for surface in self._create_surfaces(): width = surface.get_width() self.draw_aaline(surface, expected_color, (0, 0), (width - 1, 0)) for x in range(width): pos = (x, 0) self.assertEqual(surface.get_at(pos), expected_color, f"pos={pos}") def test_aaline__bounding_rect(self): """Ensures draw aaline returns the correct bounding rect. Tests lines with endpoints on and off the surface. """ line_color = pygame.Color("red") surf_color = pygame.Color("blue") width = height = 30 # Using a rect to help manage where the lines are drawn. helper_rect = pygame.Rect((0, 0), (width, height)) # Testing surfaces of different sizes. One larger than the helper_rect # and one smaller (to test lines that span the surface). for size in ((width + 5, height + 5), (width - 5, height - 5)): surface = pygame.Surface(size, 0, 32) surf_rect = surface.get_rect() # Move the helper rect to different positions to test line # endpoints on and off the surface. for pos in rect_corners_mids_and_center(surf_rect): helper_rect.center = pos for start, end in self._rect_lines(helper_rect): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_aaline(surface, line_color, start, end) # Calculating the expected_rect after the line is # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect(surface, surf_color, start) self.assertEqual(bounding_rect, expected_rect) def test_aaline__surface_clip(self): """Ensures draw aaline respects a surface's clip area.""" surfw = surfh = 30 aaline_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the aaline's pos. # Test centering the pos_rect along the clip rect's edge to allow for # drawing the aaline over the clip_rect's bounds. for center in rect_corners_mids_and_center(clip_rect): pos_rect.center = center # Get the expected points by drawing the aaline without the # clip area set. surface.set_clip(None) surface.fill(surface_color) self.draw_aaline(surface, aaline_color, pos_rect.midtop, pos_rect.midbottom) expected_pts = get_color_points(surface, surface_color, clip_rect, False) # Clear the surface and set the clip area. Redraw the aaline # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_aaline(surface, aaline_color, pos_rect.midtop, pos_rect.midbottom) surface.lock() # For possible speed up. # Check all the surface points to ensure the expected_pts # are not surface_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: self.assertNotEqual(surface.get_at(pt), surface_color, pt) else: self.assertEqual(surface.get_at(pt), surface_color, pt) surface.unlock() # Commented out to avoid cluttering the test output. Add back in if draw_py # ever fully supports drawing single aalines. # class PythonDrawAALineTest(AALineMixin, PythonDrawTestCase): # """Test draw_py module function aaline. # # This class inherits the general tests from AALineMixin. It is also the # class to add any draw_py.draw_aaline specific tests to. # """ class DrawAALineTest(AALineMixin, DrawTestCase): """Test draw module function aaline. This class inherits the general tests from AALineMixin. It is also the class to add any draw.aaline specific tests to. """ def test_aaline_endianness(self): """test color component order""" for depth in (24, 32): surface = pygame.Surface((5, 3), 0, depth) surface.fill(pygame.Color(0, 0, 0)) self.draw_aaline(surface, pygame.Color(255, 0, 0), (0, 1), (2, 1), 1) self.assertGreater(surface.get_at((1, 1)).r, 0, "there should be red here") surface.fill(pygame.Color(0, 0, 0)) self.draw_aaline(surface, pygame.Color(0, 0, 255), (0, 1), (2, 1), 1) self.assertGreater(surface.get_at((1, 1)).b, 0, "there should be blue here") def _check_antialiasing( self, from_point, to_point, should, check_points, set_endpoints=True ): """Draw a line between two points and check colors of check_points.""" if set_endpoints: should[from_point] = should[to_point] = FG_GREEN def check_one_direction(from_point, to_point, should): self.draw_aaline(self.surface, FG_GREEN, from_point, to_point, True) for pt in check_points: color = should.get(pt, BG_RED) with self.subTest(from_pt=from_point, pt=pt, to=to_point): self.assertEqual(self.surface.get_at(pt), color) # reset draw.rect(self.surface, BG_RED, (0, 0, 10, 10), 0) # it is important to test also opposite direction, the algorithm # is (#512) or was not symmetric check_one_direction(from_point, to_point, should) if from_point != to_point: check_one_direction(to_point, from_point, should) def test_short_non_antialiased_lines(self): """test very short not anti aliased lines in all directions.""" # Horizontal, vertical and diagonal lines should not be anti-aliased, # even with draw.aaline ... self.surface = pygame.Surface((10, 10)) draw.rect(self.surface, BG_RED, (0, 0, 10, 10), 0) check_points = [(i, j) for i in range(3, 8) for j in range(3, 8)] def check_both_directions(from_pt, to_pt, other_points): should = {pt: FG_GREEN for pt in other_points} self._check_antialiasing(from_pt, to_pt, should, check_points) # 0. one point check_both_directions((5, 5), (5, 5), []) # 1. horizontal check_both_directions((4, 7), (5, 7), []) check_both_directions((5, 4), (7, 4), [(6, 4)]) # 2. vertical check_both_directions((5, 5), (5, 6), []) check_both_directions((6, 4), (6, 6), [(6, 5)]) # 3. diagonals check_both_directions((5, 5), (6, 6), []) check_both_directions((5, 5), (7, 7), [(6, 6)]) check_both_directions((5, 6), (6, 5), []) check_both_directions((6, 4), (4, 6), [(5, 5)]) def test_short_line_anti_aliasing(self): self.surface = pygame.Surface((10, 10)) draw.rect(self.surface, BG_RED, (0, 0, 10, 10), 0) check_points = [(i, j) for i in range(3, 8) for j in range(3, 8)] def check_both_directions(from_pt, to_pt, should): self._check_antialiasing(from_pt, to_pt, should, check_points) brown = (127, 127, 0) reddish = (191, 63, 0) greenish = (63, 191, 0) # lets say dx = abs(x0 - x1) ; dy = abs(y0 - y1) # dy / dx = 0.5 check_both_directions((4, 4), (6, 5), {(5, 4): brown, (5, 5): brown}) check_both_directions((4, 5), (6, 4), {(5, 4): brown, (5, 5): brown}) # dy / dx = 2 check_both_directions((4, 4), (5, 6), {(4, 5): brown, (5, 5): brown}) check_both_directions((5, 4), (4, 6), {(4, 5): brown, (5, 5): brown}) # some little longer lines; so we need to check more points: check_points = [(i, j) for i in range(2, 9) for j in range(2, 9)] # dy / dx = 0.25 should = { (4, 3): greenish, (5, 3): brown, (6, 3): reddish, (4, 4): reddish, (5, 4): brown, (6, 4): greenish, } check_both_directions((3, 3), (7, 4), should) should = { (4, 3): reddish, (5, 3): brown, (6, 3): greenish, (4, 4): greenish, (5, 4): brown, (6, 4): reddish, } check_both_directions((3, 4), (7, 3), should) # dy / dx = 4 should = { (4, 4): greenish, (4, 5): brown, (4, 6): reddish, (5, 4): reddish, (5, 5): brown, (5, 6): greenish, } check_both_directions((4, 3), (5, 7), should) should = { (4, 4): reddish, (4, 5): brown, (4, 6): greenish, (5, 4): greenish, (5, 5): brown, (5, 6): reddish, } check_both_directions((5, 3), (4, 7), should) def test_anti_aliasing_float_coordinates(self): """Float coordinates should be blended smoothly.""" self.surface = pygame.Surface((10, 10)) draw.rect(self.surface, BG_RED, (0, 0, 10, 10), 0) check_points = [(i, j) for i in range(5) for j in range(5)] brown = (127, 127, 0) reddish = (191, 63, 0) greenish = (63, 191, 0) # 0. identical point : current implementation does no smoothing... expected = {(2, 2): FG_GREEN} self._check_antialiasing( (1.5, 2), (1.5, 2), expected, check_points, set_endpoints=False ) expected = {(2, 3): FG_GREEN} self._check_antialiasing( (2.49, 2.7), (2.49, 2.7), expected, check_points, set_endpoints=False ) # 1. horizontal lines # a) blend endpoints expected = {(1, 2): brown, (2, 2): FG_GREEN} self._check_antialiasing( (1.5, 2), (2, 2), expected, check_points, set_endpoints=False ) expected = {(1, 2): brown, (2, 2): FG_GREEN, (3, 2): brown} self._check_antialiasing( (1.5, 2), (2.5, 2), expected, check_points, set_endpoints=False ) expected = {(2, 2): brown, (1, 2): FG_GREEN} self._check_antialiasing( (1, 2), (1.5, 2), expected, check_points, set_endpoints=False ) expected = {(1, 2): brown, (2, 2): greenish} self._check_antialiasing( (1.5, 2), (1.75, 2), expected, check_points, set_endpoints=False ) # b) blend y-coordinate expected = {(x, y): brown for x in range(2, 5) for y in (1, 2)} self._check_antialiasing( (2, 1.5), (4, 1.5), expected, check_points, set_endpoints=False ) # 2. vertical lines # a) blend endpoints expected = {(2, 1): brown, (2, 2): FG_GREEN, (2, 3): brown} self._check_antialiasing( (2, 1.5), (2, 2.5), expected, check_points, set_endpoints=False ) expected = {(2, 1): brown, (2, 2): greenish} self._check_antialiasing( (2, 1.5), (2, 1.75), expected, check_points, set_endpoints=False ) # b) blend x-coordinate expected = {(x, y): brown for x in (1, 2) for y in range(2, 5)} self._check_antialiasing( (1.5, 2), (1.5, 4), expected, check_points, set_endpoints=False ) # 3. diagonal lines # a) blend endpoints expected = {(1, 1): brown, (2, 2): FG_GREEN, (3, 3): brown} self._check_antialiasing( (1.5, 1.5), (2.5, 2.5), expected, check_points, set_endpoints=False ) expected = {(3, 1): brown, (2, 2): FG_GREEN, (1, 3): brown} self._check_antialiasing( (2.5, 1.5), (1.5, 2.5), expected, check_points, set_endpoints=False ) # b) blend sidewards expected = {(2, 1): brown, (2, 2): brown, (3, 2): brown, (3, 3): brown} self._check_antialiasing( (2, 1.5), (3, 2.5), expected, check_points, set_endpoints=False ) expected = { (2, 1): greenish, (2, 2): reddish, (3, 2): greenish, (3, 3): reddish, (4, 3): greenish, (4, 4): reddish, } self._check_antialiasing( (2, 1.25), (4, 3.25), expected, check_points, set_endpoints=False ) def test_anti_aliasing_at_and_outside_the_border(self): """Ensures antialiasing works correct at a surface's borders.""" self.surface = pygame.Surface((10, 10)) draw.rect(self.surface, BG_RED, (0, 0, 10, 10), 0) check_points = [(i, j) for i in range(10) for j in range(10)] reddish = (191, 63, 0) brown = (127, 127, 0) greenish = (63, 191, 0) from_point, to_point = (3, 3), (7, 4) should = { (4, 3): greenish, (5, 3): brown, (6, 3): reddish, (4, 4): reddish, (5, 4): brown, (6, 4): greenish, } for dx, dy in ( (-4, 0), (4, 0), # moved to left and right borders (0, -5), (0, -4), (0, -3), # upper border (0, 5), (0, 6), (0, 7), # lower border (-4, -4), (-4, -3), (-3, -4), ): # upper left corner first = from_point[0] + dx, from_point[1] + dy second = to_point[0] + dx, to_point[1] + dy expected = {(x + dx, y + dy): color for (x, y), color in should.items()} self._check_antialiasing(first, second, expected, check_points) ### AALines Testing ########################################################### class AALinesMixin(BaseLineMixin): """Mixin test for drawing aalines. This class contains all the general aalines drawing tests. """ def test_aalines__args(self): """Ensures draw aalines accepts the correct args.""" bounds_rect = self.draw_aalines( pygame.Surface((3, 3)), (0, 10, 0, 50), False, ((0, 0), (1, 1)), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__args_without_blend(self): """Ensures draw aalines accepts the args without a blend.""" bounds_rect = self.draw_aalines( pygame.Surface((2, 2)), (0, 0, 0, 50), False, ((0, 0), (1, 1)) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__blend_warning(self): """From pygame 2, blend=False should raise DeprecationWarning.""" with warnings.catch_warnings(record=True) as w: # Cause all warnings to always be triggered. warnings.simplefilter("always") # Trigger DeprecationWarning. self.draw_aalines( pygame.Surface((2, 2)), (0, 0, 0, 50), False, ((0, 0), (1, 1)), False ) # Check if there is only one warning and is a DeprecationWarning. self.assertEqual(len(w), 1) self.assertTrue(issubclass(w[-1].category, DeprecationWarning)) def test_aalines__kwargs(self): """Ensures draw aalines accepts the correct kwargs.""" surface = pygame.Surface((4, 4)) color = pygame.Color("yellow") points = ((0, 0), (1, 1), (2, 2)) kwargs_list = [ {"surface": surface, "color": color, "closed": False, "points": points}, ] for kwargs in kwargs_list: bounds_rect = self.draw_aalines(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__kwargs_order_independent(self): """Ensures draw aalines's kwargs are not order dependent.""" bounds_rect = self.draw_aalines( closed=1, points=((0, 0), (1, 1), (2, 2)), color=(10, 20, 30), surface=pygame.Surface((3, 2)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__args_missing(self): """Ensures draw aalines detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(surface, color, 0) with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_aalines() def test_aalines__kwargs_missing(self): """Ensures draw aalines detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((3, 2)), "color": pygame.Color("red"), "closed": 1, "points": ((2, 2), (1, 1)), } for name in ("points", "closed", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(**invalid_kwargs) def test_aalines__arg_invalid_types(self): """Ensures draw aalines detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") closed = 0 points = ((1, 2), (2, 1)) with self.assertRaises(TypeError): # Invalid blend. bounds_rect = self.draw_aalines(surface, color, closed, points, "1") with self.assertRaises(TypeError): # Invalid points. bounds_rect = self.draw_aalines(surface, color, closed, (1, 2, 3)) with self.assertRaises(TypeError): # Invalid closed. bounds_rect = self.draw_aalines(surface, color, InvalidBool(), points) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_aalines(surface, 2.3, closed, points) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_aalines((1, 2, 3, 4), color, closed, points) def test_aalines__kwarg_invalid_types(self): """Ensures draw aalines detects invalid kwarg types.""" valid_kwargs = { "surface": pygame.Surface((3, 3)), "color": pygame.Color("green"), "closed": False, "points": ((1, 2), (2, 1)), } invalid_kwargs = { "surface": pygame.Surface, "color": 2.3, "closed": InvalidBool(), "points": (0, 0, 0), } for kwarg in ("surface", "color", "closed", "points"): kwargs = dict(valid_kwargs) kwargs[kwarg] = invalid_kwargs[kwarg] with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(**kwargs) def test_aalines__kwarg_invalid_name(self): """Ensures draw aalines detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") closed = 1 points = ((1, 2), (2, 1)) kwargs_list = [ { "surface": surface, "color": color, "closed": closed, "points": points, "invalid": 1, }, { "surface": surface, "color": color, "closed": closed, "points": points, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(**kwargs) def test_aalines__args_and_kwargs(self): """Ensures draw aalines accepts a combination of args/kwargs""" surface = pygame.Surface((3, 2)) color = (255, 255, 0, 0) closed = 0 points = ((1, 2), (2, 1)) kwargs = { "surface": surface, "color": color, "closed": closed, "points": points, } for name in ("surface", "color", "closed", "points"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_aalines(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_aalines(surface, color, **kwargs) elif "closed" == name: bounds_rect = self.draw_aalines(surface, color, closed, **kwargs) elif "points" == name: bounds_rect = self.draw_aalines( surface, color, closed, points, **kwargs ) else: bounds_rect = self.draw_aalines( surface, color, closed, points, **kwargs ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__valid_points_format(self): """Ensures draw aalines accepts different points formats.""" expected_color = (10, 20, 30, 255) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": expected_color, "closed": False, "points": None, } # The point type can be a tuple/list/Vector2. point_types = ( (tuple, tuple, tuple, tuple), # all tuples (list, list, list, list), # all lists (Vector2, Vector2, Vector2, Vector2), # all Vector2s (list, Vector2, tuple, Vector2), ) # mix # The point values can be ints or floats. point_values = ( ((1, 1), (2, 1), (2, 2), (1, 2)), ((1, 1), (2.2, 1), (2.1, 2.2), (1, 2.1)), ) # Each sequence of points can be a tuple or a list. seq_types = (tuple, list) for point_type in point_types: for values in point_values: check_pos = values[0] points = [point_type[i](pt) for i, pt in enumerate(values)] for seq_type in seq_types: surface.fill(surface_color) # Clear for each test. kwargs["points"] = seq_type(points) bounds_rect = self.draw_aalines(**kwargs) self.assertEqual(surface.get_at(check_pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__invalid_points_formats(self): """Ensures draw aalines handles invalid points formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "closed": False, "points": None, } points_fmts = ( ((1, 1), (2,)), # Too few coords. ((1, 1), (2, 2, 2)), # Too many coords. ((1, 1), (2, "2")), # Wrong type. ((1, 1), {2, 3}), # Wrong type. ((1, 1), dict(((2, 2), (3, 3)))), # Wrong type. {(1, 1), (1, 2)}, # Wrong type. dict(((1, 1), (4, 4))), ) # Wrong type. for points in points_fmts: kwargs["points"] = points with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(**kwargs) def test_aalines__invalid_points_values(self): """Ensures draw aalines handles invalid points values correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "closed": False, "points": None, } for points in ([], ((1, 1),)): # Too few points. for seq_type in (tuple, list): # Test as tuples and lists. kwargs["points"] = seq_type(points) with self.assertRaises(ValueError): bounds_rect = self.draw_aalines(**kwargs) def test_aalines__valid_closed_values(self): """Ensures draw aalines accepts different closed values.""" line_color = pygame.Color("blue") surface_color = pygame.Color("white") surface = pygame.Surface((5, 5)) pos = (1, 3) kwargs = { "surface": surface, "color": line_color, "closed": None, "points": ((1, 1), (4, 1), (4, 4), (1, 4)), } true_values = (-7, 1, 10, "2", 3.1, (4,), [5], True) false_values = (None, "", 0, (), [], False) for closed in true_values + false_values: surface.fill(surface_color) # Clear for each test. kwargs["closed"] = closed expected_color = line_color if closed else surface_color bounds_rect = self.draw_aalines(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__valid_color_formats(self): """Ensures draw aalines accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) pos = (1, 1) kwargs = { "surface": surface, "color": None, "closed": False, "points": (pos, (2, 1)), } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_aalines(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_aalines__invalid_color_formats(self): """Ensures draw aalines handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "closed": False, "points": ((1, 1), (1, 2)), } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_aalines(**kwargs) def test_aalines__color(self): """Tests if the aalines drawn are the correct color. Draws aalines around the border of the given surface and checks if all borders of the surface only contain the given color. """ for surface in self._create_surfaces(): for expected_color in self.COLORS: self.draw_aalines(surface, expected_color, True, corners(surface)) for pos, color in border_pos_and_color(surface): self.assertEqual(color, expected_color, f"pos={pos}") def test_aalines__gaps(self): """Tests if the aalines drawn contain any gaps. Draws aalines around the border of the given surface and checks if all borders of the surface contain any gaps. See: #512 """ expected_color = (255, 255, 255) for surface in self._create_surfaces(): self.draw_aalines(surface, expected_color, True, corners(surface)) for pos, color in border_pos_and_color(surface): self.assertEqual(color, expected_color, f"pos={pos}") def test_aalines__bounding_rect(self): """Ensures draw aalines returns the correct bounding rect. Tests lines with endpoints on and off the surface and blending enabled and disabled. """ line_color = pygame.Color("red") surf_color = pygame.Color("blue") width = height = 30 # Using a rect to help manage where the lines are drawn. pos_rect = pygame.Rect((0, 0), (width, height)) # Testing surfaces of different sizes. One larger than the pos_rect # and one smaller (to test lines that span the surface). for size in ((width + 5, height + 5), (width - 5, height - 5)): surface = pygame.Surface(size, 0, 32) surf_rect = surface.get_rect() # Move pos_rect to different positions to test line endpoints on # and off the surface. for pos in rect_corners_mids_and_center(surf_rect): pos_rect.center = pos # Shape: Triangle (if closed), ^ caret (if not closed). pts = (pos_rect.midleft, pos_rect.midtop, pos_rect.midright) pos = pts[0] # Rect position if nothing drawn. for closed in (True, False): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_aalines(surface, line_color, closed, pts) # Calculating the expected_rect after the lines are # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect(surface, surf_color, pos) self.assertEqual(bounding_rect, expected_rect) def test_aalines__surface_clip(self): """Ensures draw aalines respects a surface's clip area.""" surfw = surfh = 30 aaline_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the aalines's pos. # Test centering the pos_rect along the clip rect's edge to allow for # drawing the aalines over the clip_rect's bounds. for center in rect_corners_mids_and_center(clip_rect): pos_rect.center = center pts = (pos_rect.midtop, pos_rect.center, pos_rect.midbottom) for closed in (True, False): # Test closed and not closed. # Get the expected points by drawing the aalines without # the clip area set. surface.set_clip(None) surface.fill(surface_color) self.draw_aalines(surface, aaline_color, closed, pts) expected_pts = get_color_points( surface, surface_color, clip_rect, False ) # Clear the surface and set the clip area. Redraw the # aalines and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_aalines(surface, aaline_color, closed, pts) surface.lock() # For possible speed up. # Check all the surface points to ensure the expected_pts # are not surface_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: self.assertNotEqual(surface.get_at(pt), surface_color, pt) else: self.assertEqual(surface.get_at(pt), surface_color, pt) surface.unlock() # Commented out to avoid cluttering the test output. Add back in if draw_py # ever fully supports drawing aalines. # class PythonDrawAALinesTest(AALinesMixin, PythonDrawTestCase): # """Test draw_py module function aalines. # # This class inherits the general tests from AALinesMixin. It is also the # class to add any draw_py.draw_aalines specific tests to. # """ class DrawAALinesTest(AALinesMixin, DrawTestCase): """Test draw module function aalines. This class inherits the general tests from AALinesMixin. It is also the class to add any draw.aalines specific tests to. """ ### Polygon Testing ########################################################### SQUARE = ([0, 0], [3, 0], [3, 3], [0, 3]) DIAMOND = [(1, 3), (3, 5), (5, 3), (3, 1)] CROSS = ( [2, 0], [4, 0], [4, 2], [6, 2], [6, 4], [4, 4], [4, 6], [2, 6], [2, 4], [0, 4], [0, 2], [2, 2], ) class DrawPolygonMixin: """Mixin tests for drawing polygons. This class contains all the general polygon drawing tests. """ def setUp(self): self.surface = pygame.Surface((20, 20)) def test_polygon__args(self): """Ensures draw polygon accepts the correct args.""" bounds_rect = self.draw_polygon( pygame.Surface((3, 3)), (0, 10, 0, 50), ((0, 0), (1, 1), (2, 2)), 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__args_without_width(self): """Ensures draw polygon accepts the args without a width.""" bounds_rect = self.draw_polygon( pygame.Surface((2, 2)), (0, 0, 0, 50), ((0, 0), (1, 1), (2, 2)) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__kwargs(self): """Ensures draw polygon accepts the correct kwargs with and without a width arg. """ surface = pygame.Surface((4, 4)) color = pygame.Color("yellow") points = ((0, 0), (1, 1), (2, 2)) kwargs_list = [ {"surface": surface, "color": color, "points": points, "width": 1}, {"surface": surface, "color": color, "points": points}, ] for kwargs in kwargs_list: bounds_rect = self.draw_polygon(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__kwargs_order_independent(self): """Ensures draw polygon's kwargs are not order dependent.""" bounds_rect = self.draw_polygon( color=(10, 20, 30), surface=pygame.Surface((3, 2)), width=0, points=((0, 1), (1, 2), (2, 3)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__args_missing(self): """Ensures draw polygon detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_polygon() def test_polygon__kwargs_missing(self): """Ensures draw polygon detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((1, 2)), "color": pygame.Color("red"), "points": ((2, 1), (2, 2), (2, 3)), "width": 1, } for name in ("points", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(**invalid_kwargs) def test_polygon__arg_invalid_types(self): """Ensures draw polygon detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") points = ((0, 1), (1, 2), (1, 3)) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_polygon(surface, color, points, "1") with self.assertRaises(TypeError): # Invalid points. bounds_rect = self.draw_polygon(surface, color, (1, 2, 3)) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_polygon(surface, 2.3, points) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_polygon((1, 2, 3, 4), color, points) def test_polygon__kwarg_invalid_types(self): """Ensures draw polygon detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") points = ((0, 0), (1, 0), (2, 0)) width = 1 kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "points": points, "width": width, }, { "surface": surface, "color": 2.3, # Invalid color. "points": points, "width": width, }, { "surface": surface, "color": color, "points": ((1,), (1,), (1,)), # Invalid points. "width": width, }, {"surface": surface, "color": color, "points": points, "width": 1.2}, ] # Invalid width. for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(**kwargs) def test_polygon__kwarg_invalid_name(self): """Ensures draw polygon detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") points = ((1, 1), (1, 2), (1, 3)) kwargs_list = [ { "surface": surface, "color": color, "points": points, "width": 1, "invalid": 1, }, {"surface": surface, "color": color, "points": points, "invalid": 1}, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(**kwargs) def test_polygon__args_and_kwargs(self): """Ensures draw polygon accepts a combination of args/kwargs""" surface = pygame.Surface((3, 1)) color = (255, 255, 0, 0) points = ((0, 1), (1, 2), (2, 3)) width = 0 kwargs = {"surface": surface, "color": color, "points": points, "width": width} for name in ("surface", "color", "points", "width"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_polygon(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_polygon(surface, color, **kwargs) elif "points" == name: bounds_rect = self.draw_polygon(surface, color, points, **kwargs) else: bounds_rect = self.draw_polygon(surface, color, points, width, **kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__valid_width_values(self): """Ensures draw polygon accepts different width values.""" surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) color = (10, 20, 30, 255) kwargs = { "surface": surface, "color": color, "points": ((1, 1), (2, 1), (2, 2), (1, 2)), "width": None, } pos = kwargs["points"][0] for width in (-100, -10, -1, 0, 1, 10, 100): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = color if width >= 0 else surface_color bounds_rect = self.draw_polygon(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__valid_points_format(self): """Ensures draw polygon accepts different points formats.""" expected_color = (10, 20, 30, 255) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": expected_color, "points": None, "width": 0, } # The point type can be a tuple/list/Vector2. point_types = ( (tuple, tuple, tuple, tuple), # all tuples (list, list, list, list), # all lists (Vector2, Vector2, Vector2, Vector2), # all Vector2s (list, Vector2, tuple, Vector2), ) # mix # The point values can be ints or floats. point_values = ( ((1, 1), (2, 1), (2, 2), (1, 2)), ((1, 1), (2.2, 1), (2.1, 2.2), (1, 2.1)), ) # Each sequence of points can be a tuple or a list. seq_types = (tuple, list) for point_type in point_types: for values in point_values: check_pos = values[0] points = [point_type[i](pt) for i, pt in enumerate(values)] for seq_type in seq_types: surface.fill(surface_color) # Clear for each test. kwargs["points"] = seq_type(points) bounds_rect = self.draw_polygon(**kwargs) self.assertEqual(surface.get_at(check_pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__invalid_points_formats(self): """Ensures draw polygon handles invalid points formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "points": None, "width": 0, } points_fmts = ( ((1, 1), (2, 1), (2,)), # Too few coords. ((1, 1), (2, 1), (2, 2, 2)), # Too many coords. ((1, 1), (2, 1), (2, "2")), # Wrong type. ((1, 1), (2, 1), {2, 3}), # Wrong type. ((1, 1), (2, 1), dict(((2, 2), (3, 3)))), # Wrong type. {(1, 1), (2, 1), (2, 2), (1, 2)}, # Wrong type. dict(((1, 1), (2, 2), (3, 3), (4, 4))), ) # Wrong type. for points in points_fmts: kwargs["points"] = points with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(**kwargs) def test_polygon__invalid_points_values(self): """Ensures draw polygon handles invalid points values correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "points": None, "width": 0, } points_fmts = ( tuple(), # Too few points. ((1, 1),), # Too few points. ((1, 1), (2, 1)), ) # Too few points. for points in points_fmts: for seq_type in (tuple, list): # Test as tuples and lists. kwargs["points"] = seq_type(points) with self.assertRaises(ValueError): bounds_rect = self.draw_polygon(**kwargs) def test_polygon__valid_color_formats(self): """Ensures draw polygon accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": None, "points": ((1, 1), (2, 1), (2, 2), (1, 2)), "width": 0, } pos = kwargs["points"][0] greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_polygon(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_polygon__invalid_color_formats(self): """Ensures draw polygon handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "points": ((1, 1), (2, 1), (2, 2), (1, 2)), "width": 0, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_polygon(**kwargs) def test_draw_square(self): self.draw_polygon(self.surface, RED, SQUARE, 0) # note : there is a discussion (#234) if draw.polygon should include or # not the right or lower border; here we stick with current behavior, # eg include those borders ... for x in range(4): for y in range(4): self.assertEqual(self.surface.get_at((x, y)), RED) def test_draw_diamond(self): pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) self.draw_polygon(self.surface, GREEN, DIAMOND, 0) # this diamond shape is equivalent to its four corners, plus inner square for x, y in DIAMOND: self.assertEqual(self.surface.get_at((x, y)), GREEN, msg=str((x, y))) for x in range(2, 5): for y in range(2, 5): self.assertEqual(self.surface.get_at((x, y)), GREEN) def test_1_pixel_high_or_wide_shapes(self): # 1. one-pixel-high, filled pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) self.draw_polygon(self.surface, GREEN, [(x, 2) for x, _y in CROSS], 0) cross_size = 6 # the maximum x or y coordinate of the cross for x in range(cross_size + 1): self.assertEqual(self.surface.get_at((x, 1)), RED) self.assertEqual(self.surface.get_at((x, 2)), GREEN) self.assertEqual(self.surface.get_at((x, 3)), RED) pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) # 2. one-pixel-high, not filled self.draw_polygon(self.surface, GREEN, [(x, 5) for x, _y in CROSS], 1) for x in range(cross_size + 1): self.assertEqual(self.surface.get_at((x, 4)), RED) self.assertEqual(self.surface.get_at((x, 5)), GREEN) self.assertEqual(self.surface.get_at((x, 6)), RED) pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) # 3. one-pixel-wide, filled self.draw_polygon(self.surface, GREEN, [(3, y) for _x, y in CROSS], 0) for y in range(cross_size + 1): self.assertEqual(self.surface.get_at((2, y)), RED) self.assertEqual(self.surface.get_at((3, y)), GREEN) self.assertEqual(self.surface.get_at((4, y)), RED) pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) # 4. one-pixel-wide, not filled self.draw_polygon(self.surface, GREEN, [(4, y) for _x, y in CROSS], 1) for y in range(cross_size + 1): self.assertEqual(self.surface.get_at((3, y)), RED) self.assertEqual(self.surface.get_at((4, y)), GREEN) self.assertEqual(self.surface.get_at((5, y)), RED) def test_draw_symetric_cross(self): """non-regression on issue #234 : x and y where handled inconsistently. Also, the result is/was different whether we fill or not the polygon. """ # 1. case width = 1 (not filled: `polygon` calls internally the `lines` function) pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) self.draw_polygon(self.surface, GREEN, CROSS, 1) inside = [(x, 3) for x in range(1, 6)] + [(3, y) for y in range(1, 6)] for x in range(10): for y in range(10): if (x, y) in inside: self.assertEqual(self.surface.get_at((x, y)), RED) elif (x in range(2, 5) and y < 7) or (y in range(2, 5) and x < 7): # we are on the border of the cross: self.assertEqual(self.surface.get_at((x, y)), GREEN) else: # we are outside self.assertEqual(self.surface.get_at((x, y)), RED) # 2. case width = 0 (filled; this is the example from #234) pygame.draw.rect(self.surface, RED, (0, 0, 10, 10), 0) self.draw_polygon(self.surface, GREEN, CROSS, 0) inside = [(x, 3) for x in range(1, 6)] + [(3, y) for y in range(1, 6)] for x in range(10): for y in range(10): if (x in range(2, 5) and y < 7) or (y in range(2, 5) and x < 7): # we are on the border of the cross: self.assertEqual( self.surface.get_at((x, y)), GREEN, msg=str((x, y)) ) else: # we are outside self.assertEqual(self.surface.get_at((x, y)), RED) def test_illumine_shape(self): """non-regression on issue #313""" rect = pygame.Rect((0, 0, 20, 20)) path_data = [ (0, 0), (rect.width - 1, 0), # upper border (rect.width - 5, 5 - 1), (5 - 1, 5 - 1), # upper inner (5 - 1, rect.height - 5), (0, rect.height - 1), ] # lower diagonal # The shape looks like this (the numbers are the indices of path_data) # 0**********************1 <-- upper border # *********************** # ********************** # ********************* # ****3**************2 <-- upper inner border # ***** # ***** (more lines here) # ***** # ****4 # **** # *** # ** # 5 # # the current bug is that the "upper inner" line is not drawn, but only # if 4 or some lower corner exists pygame.draw.rect(self.surface, RED, (0, 0, 20, 20), 0) # 1. First without the corners 4 & 5 self.draw_polygon(self.surface, GREEN, path_data[:4], 0) for x in range(20): self.assertEqual(self.surface.get_at((x, 0)), GREEN) # upper border for x in range(4, rect.width - 5 + 1): self.assertEqual(self.surface.get_at((x, 4)), GREEN) # upper inner # 2. with the corners 4 & 5 pygame.draw.rect(self.surface, RED, (0, 0, 20, 20), 0) self.draw_polygon(self.surface, GREEN, path_data, 0) for x in range(4, rect.width - 5 + 1): self.assertEqual(self.surface.get_at((x, 4)), GREEN) # upper inner def test_invalid_points(self): self.assertRaises( TypeError, lambda: self.draw_polygon( self.surface, RED, ((0, 0), (0, 20), (20, 20), 20), 0 ), ) def test_polygon__bounding_rect(self): """Ensures draw polygon returns the correct bounding rect. Tests polygons on and off the surface and a range of width/thickness values. """ polygon_color = pygame.Color("red") surf_color = pygame.Color("black") min_width = min_height = 5 max_width = max_height = 7 sizes = ((min_width, min_height), (max_width, max_height)) surface = pygame.Surface((20, 20), 0, 32) surf_rect = surface.get_rect() # Make a rect that is bigger than the surface to help test drawing # polygons off and partially off the surface. big_rect = surf_rect.inflate(min_width * 2 + 1, min_height * 2 + 1) for pos in rect_corners_mids_and_center( surf_rect ) + rect_corners_mids_and_center(big_rect): # A rect (pos_rect) is used to help create and position the # polygon. Each of this rect's position attributes will be set to # the pos value. for attr in RECT_POSITION_ATTRIBUTES: # Test using different rect sizes and thickness values. for width, height in sizes: pos_rect = pygame.Rect((0, 0), (width, height)) setattr(pos_rect, attr, pos) # Points form a triangle with no fully # horizontal/vertical lines. vertices = ( pos_rect.midleft, pos_rect.midtop, pos_rect.bottomright, ) for thickness in range(4): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_polygon( surface, polygon_color, vertices, thickness ) # Calculating the expected_rect after the polygon # is drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, vertices[0] ) self.assertEqual( bounding_rect, expected_rect, f"thickness={thickness}", ) def test_polygon__surface_clip(self): """Ensures draw polygon respects a surface's clip area. Tests drawing the polygon filled and unfilled. """ surfw = surfh = 30 polygon_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (8, 10)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the polygon's pos. for width in (0, 1): # Filled and unfilled. # Test centering the polygon along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the polygon without the # clip area set. pos_rect.center = center vertices = ( pos_rect.topleft, pos_rect.topright, pos_rect.bottomright, pos_rect.bottomleft, ) surface.set_clip(None) surface.fill(surface_color) self.draw_polygon(surface, polygon_color, vertices, width) expected_pts = get_color_points(surface, polygon_color, clip_rect) # Clear the surface and set the clip area. Redraw the polygon # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_polygon(surface, polygon_color, vertices, width) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the polygon_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = polygon_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() class DrawPolygonTest(DrawPolygonMixin, DrawTestCase): """Test draw module function polygon. This class inherits the general tests from DrawPolygonMixin. It is also the class to add any draw.polygon specific tests to. """ # Commented out to avoid cluttering the test output. Add back in if draw_py # ever fully supports drawing polygons. # @unittest.skip('draw_py.draw_polygon not fully supported yet') # class PythonDrawPolygonTest(DrawPolygonMixin, PythonDrawTestCase): # """Test draw_py module function draw_polygon. # # This class inherits the general tests from DrawPolygonMixin. It is also # the class to add any draw_py.draw_polygon specific tests to. # """ ### Rect Testing ############################################################## class DrawRectMixin: """Mixin tests for drawing rects. This class contains all the general rect drawing tests. """ def test_rect__args(self): """Ensures draw rect accepts the correct args.""" bounds_rect = self.draw_rect( pygame.Surface((2, 2)), (20, 10, 20, 150), pygame.Rect((0, 0), (1, 1)), 2, 1, 2, 3, 4, 5, ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__args_without_width(self): """Ensures draw rect accepts the args without a width and borders.""" bounds_rect = self.draw_rect( pygame.Surface((3, 5)), (0, 0, 0, 255), pygame.Rect((0, 0), (1, 1)) ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__kwargs(self): """Ensures draw rect accepts the correct kwargs with and without a width and border_radius arg. """ kwargs_list = [ { "surface": pygame.Surface((5, 5)), "color": pygame.Color("red"), "rect": pygame.Rect((0, 0), (1, 2)), "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": pygame.Surface((1, 2)), "color": (0, 100, 200), "rect": (0, 0, 1, 1), }, ] for kwargs in kwargs_list: bounds_rect = self.draw_rect(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__kwargs_order_independent(self): """Ensures draw rect's kwargs are not order dependent.""" bounds_rect = self.draw_rect( color=(0, 1, 2), border_radius=10, surface=pygame.Surface((2, 3)), border_top_left_radius=5, width=-2, border_top_right_radius=20, border_bottom_right_radius=0, rect=pygame.Rect((0, 0), (0, 0)), border_bottom_left_radius=15, ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__args_missing(self): """Ensures draw rect detects any missing required args.""" surface = pygame.Surface((1, 1)) with self.assertRaises(TypeError): bounds_rect = self.draw_rect(surface, pygame.Color("white")) with self.assertRaises(TypeError): bounds_rect = self.draw_rect(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_rect() def test_rect__kwargs_missing(self): """Ensures draw rect detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((1, 3)), "color": pygame.Color("red"), "rect": pygame.Rect((0, 0), (2, 2)), "width": 5, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, } for name in ("rect", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_rect(**invalid_kwargs) def test_rect__arg_invalid_types(self): """Ensures draw rect detects invalid arg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("white") rect = pygame.Rect((1, 1), (1, 1)) with self.assertRaises(TypeError): # Invalid border_bottom_right_radius. bounds_rect = self.draw_rect( surface, color, rect, 2, border_bottom_right_radius="rad" ) with self.assertRaises(TypeError): # Invalid border_bottom_left_radius. bounds_rect = self.draw_rect( surface, color, rect, 2, border_bottom_left_radius="rad" ) with self.assertRaises(TypeError): # Invalid border_top_right_radius. bounds_rect = self.draw_rect( surface, color, rect, 2, border_top_right_radius="rad" ) with self.assertRaises(TypeError): # Invalid border_top_left_radius. bounds_rect = self.draw_rect( surface, color, rect, 2, border_top_left_radius="draw" ) with self.assertRaises(TypeError): # Invalid border_radius. bounds_rect = self.draw_rect(surface, color, rect, 2, "rad") with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_rect(surface, color, rect, "2", 4) with self.assertRaises(TypeError): # Invalid rect. bounds_rect = self.draw_rect(surface, color, (1, 2, 3), 2, 6) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_rect(surface, 2.3, rect, 3, 8) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_rect(rect, color, rect, 4, 10) def test_rect__kwarg_invalid_types(self): """Ensures draw rect detects invalid kwarg types.""" surface = pygame.Surface((2, 3)) color = pygame.Color("red") rect = pygame.Rect((0, 0), (1, 1)) kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": 2.3, # Invalid color. "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": (1, 1, 2), # Invalid rect. "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1.1, # Invalid width. "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10.5, # Invalid border_radius. "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5.5, # Invalid top_left_radius. "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": "a", # Invalid top_right_radius. "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": "c", # Invalid bottom_left_radius "border_bottom_right_radius": 0, }, { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": "d", # Invalid bottom_right. }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_rect(**kwargs) def test_rect__kwarg_invalid_name(self): """Ensures draw rect detects invalid kwarg names.""" surface = pygame.Surface((2, 1)) color = pygame.Color("green") rect = pygame.Rect((0, 0), (3, 3)) kwargs_list = [ { "surface": surface, "color": color, "rect": rect, "width": 1, "border_radius": 10, "border_top_left_radius": 5, "border_top_right_radius": 20, "border_bottom_left_radius": 15, "border_bottom_right_radius": 0, "invalid": 1, }, {"surface": surface, "color": color, "rect": rect, "invalid": 1}, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_rect(**kwargs) def test_rect__args_and_kwargs(self): """Ensures draw rect accepts a combination of args/kwargs""" surface = pygame.Surface((3, 1)) color = (255, 255, 255, 0) rect = pygame.Rect((1, 0), (2, 5)) width = 0 kwargs = {"surface": surface, "color": color, "rect": rect, "width": width} for name in ("surface", "color", "rect", "width"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_rect(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_rect(surface, color, **kwargs) elif "rect" == name: bounds_rect = self.draw_rect(surface, color, rect, **kwargs) else: bounds_rect = self.draw_rect(surface, color, rect, width, **kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__valid_width_values(self): """Ensures draw rect accepts different width values.""" pos = (1, 1) surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) color = (1, 2, 3, 255) kwargs = { "surface": surface, "color": color, "rect": pygame.Rect(pos, (2, 2)), "width": None, } for width in (-1000, -10, -1, 0, 1, 10, 1000): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = color if width >= 0 else surface_color bounds_rect = self.draw_rect(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__valid_rect_formats(self): """Ensures draw rect accepts different rect formats.""" pos = (1, 1) expected_color = pygame.Color("yellow") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) kwargs = {"surface": surface, "color": expected_color, "rect": None, "width": 0} rects = ( pygame.Rect(pos, (1, 1)), (pos, (2, 2)), (pos[0], pos[1], 3, 3), [pos, (2.1, 2.2)], ) for rect in rects: surface.fill(surface_color) # Clear for each test. kwargs["rect"] = rect bounds_rect = self.draw_rect(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__invalid_rect_formats(self): """Ensures draw rect handles invalid rect formats correctly.""" kwargs = { "surface": pygame.Surface((4, 4)), "color": pygame.Color("red"), "rect": None, "width": 0, } invalid_fmts = ( [], [1], [1, 2], [1, 2, 3], [1, 2, 3, 4, 5], {1, 2, 3, 4}, [1, 2, 3, "4"], ) for rect in invalid_fmts: kwargs["rect"] = rect with self.assertRaises(TypeError): bounds_rect = self.draw_rect(**kwargs) def test_rect__valid_color_formats(self): """Ensures draw rect accepts different color formats.""" pos = (1, 1) red_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": None, "rect": pygame.Rect(pos, (1, 1)), "width": 3, } reds = ((255, 0, 0), (255, 0, 0, 255), surface.map_rgb(red_color), red_color) for color in reds: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = red_color bounds_rect = self.draw_rect(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_rect__invalid_color_formats(self): """Ensures draw rect handles invalid color formats correctly.""" pos = (1, 1) surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": None, "rect": pygame.Rect(pos, (1, 1)), "width": 1, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_rect(**kwargs) def test_rect__fill(self): self.surf_w, self.surf_h = self.surf_size = (320, 200) self.surf = pygame.Surface(self.surf_size, pygame.SRCALPHA) self.color = (1, 13, 24, 205) rect = pygame.Rect(10, 10, 25, 20) drawn = self.draw_rect(self.surf, self.color, rect, 0) self.assertEqual(drawn, rect) # Should be colored where it's supposed to be for pt in test_utils.rect_area_pts(rect): color_at_pt = self.surf.get_at(pt) self.assertEqual(color_at_pt, self.color) # And not where it shouldn't for pt in test_utils.rect_outer_bounds(rect): color_at_pt = self.surf.get_at(pt) self.assertNotEqual(color_at_pt, self.color) # Issue #310: Cannot draw rectangles that are 1 pixel high bgcolor = pygame.Color("black") self.surf.fill(bgcolor) hrect = pygame.Rect(1, 1, self.surf_w - 2, 1) vrect = pygame.Rect(1, 3, 1, self.surf_h - 4) drawn = self.draw_rect(self.surf, self.color, hrect, 0) self.assertEqual(drawn, hrect) x, y = hrect.topleft w, h = hrect.size self.assertEqual(self.surf.get_at((x - 1, y)), bgcolor) self.assertEqual(self.surf.get_at((x + w, y)), bgcolor) for i in range(x, x + w): self.assertEqual(self.surf.get_at((i, y)), self.color) drawn = self.draw_rect(self.surf, self.color, vrect, 0) self.assertEqual(drawn, vrect) x, y = vrect.topleft w, h = vrect.size self.assertEqual(self.surf.get_at((x, y - 1)), bgcolor) self.assertEqual(self.surf.get_at((x, y + h)), bgcolor) for i in range(y, y + h): self.assertEqual(self.surf.get_at((x, i)), self.color) def test_rect__one_pixel_lines(self): self.surf = pygame.Surface((320, 200), pygame.SRCALPHA) self.color = (1, 13, 24, 205) rect = pygame.Rect(10, 10, 56, 20) drawn = self.draw_rect(self.surf, self.color, rect, 1) self.assertEqual(drawn, rect) # Should be colored where it's supposed to be for pt in test_utils.rect_perimeter_pts(drawn): color_at_pt = self.surf.get_at(pt) self.assertEqual(color_at_pt, self.color) # And not where it shouldn't for pt in test_utils.rect_outer_bounds(drawn): color_at_pt = self.surf.get_at(pt) self.assertNotEqual(color_at_pt, self.color) def test_rect__draw_line_width(self): surface = pygame.Surface((100, 100)) surface.fill("black") color = pygame.Color(255, 255, 255) rect_width = 80 rect_height = 50 line_width = 10 pygame.draw.rect( surface, color, pygame.Rect(0, 0, rect_width, rect_height), line_width ) for i in range(line_width): self.assertEqual(surface.get_at((i, i)), color) self.assertEqual(surface.get_at((rect_width - i - 1, i)), color) self.assertEqual(surface.get_at((i, rect_height - i - 1)), color) self.assertEqual( surface.get_at((rect_width - i - 1, rect_height - i - 1)), color ) self.assertEqual(surface.get_at((line_width, line_width)), (0, 0, 0)) self.assertEqual( surface.get_at((rect_width - line_width - 1, line_width)), (0, 0, 0) ) self.assertEqual( surface.get_at((line_width, rect_height - line_width - 1)), (0, 0, 0) ) self.assertEqual( surface.get_at((rect_width - line_width - 1, rect_height - line_width - 1)), (0, 0, 0), ) def test_rect__bounding_rect(self): """Ensures draw rect returns the correct bounding rect. Tests rects on and off the surface and a range of width/thickness values. """ rect_color = pygame.Color("red") surf_color = pygame.Color("black") min_width = min_height = 5 max_width = max_height = 7 sizes = ((min_width, min_height), (max_width, max_height)) surface = pygame.Surface((20, 20), 0, 32) surf_rect = surface.get_rect() # Make a rect that is bigger than the surface to help test drawing # rects off and partially off the surface. big_rect = surf_rect.inflate(min_width * 2 + 1, min_height * 2 + 1) for pos in rect_corners_mids_and_center( surf_rect ) + rect_corners_mids_and_center(big_rect): # Each of the rect's position attributes will be set to the pos # value. for attr in RECT_POSITION_ATTRIBUTES: # Test using different rect sizes and thickness values. for width, height in sizes: rect = pygame.Rect((0, 0), (width, height)) setattr(rect, attr, pos) for thickness in range(4): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_rect( surface, rect_color, rect, thickness ) # Calculating the expected_rect after the rect is # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, rect.topleft ) self.assertEqual( bounding_rect, expected_rect, f"thickness={thickness}", ) def test_rect__surface_clip(self): """Ensures draw rect respects a surface's clip area. Tests drawing the rect filled and unfilled. """ surfw = surfh = 30 rect_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (8, 10)) clip_rect.center = surface.get_rect().center test_rect = clip_rect.copy() # Manages the rect's pos. for width in (0, 1): # Filled and unfilled. # Test centering the rect along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the rect without the # clip area set. test_rect.center = center surface.set_clip(None) surface.fill(surface_color) self.draw_rect(surface, rect_color, test_rect, width) expected_pts = get_color_points(surface, rect_color, clip_rect) # Clear the surface and set the clip area. Redraw the rect # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_rect(surface, rect_color, test_rect, width) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the rect_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = rect_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() class DrawRectTest(DrawRectMixin, DrawTestCase): """Test draw module function rect. This class inherits the general tests from DrawRectMixin. It is also the class to add any draw.rect specific tests to. """ # Commented out to avoid cluttering the test output. Add back in if draw_py # ever properly supports drawing rects. # @unittest.skip('draw_py.draw_rect not supported yet') # class PythonDrawRectTest(DrawRectMixin, PythonDrawTestCase): # """Test draw_py module function draw_rect. # # This class inherits the general tests from DrawRectMixin. It is also the # class to add any draw_py.draw_rect specific tests to. # """ ### Circle Testing ############################################################ class DrawCircleMixin: """Mixin tests for drawing circles. This class contains all the general circle drawing tests. """ def test_circle__args(self): """Ensures draw circle accepts the correct args.""" bounds_rect = self.draw_circle( pygame.Surface((3, 3)), (0, 10, 0, 50), (0, 0), 3, 1, 1, 0, 1, 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__args_without_width(self): """Ensures draw circle accepts the args without a width and quadrants.""" bounds_rect = self.draw_circle(pygame.Surface((2, 2)), (0, 0, 0, 50), (1, 1), 1) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__args_with_negative_width(self): """Ensures draw circle accepts the args with negative width.""" bounds_rect = self.draw_circle( pygame.Surface((2, 2)), (0, 0, 0, 50), (1, 1), 1, -1 ) self.assertIsInstance(bounds_rect, pygame.Rect) self.assertEqual(bounds_rect, pygame.Rect(1, 1, 0, 0)) def test_circle__args_with_width_gt_radius(self): """Ensures draw circle accepts the args with width > radius.""" bounds_rect = self.draw_circle( pygame.Surface((2, 2)), (0, 0, 0, 50), (1, 1), 2, 3, 0, 0, 0, 0 ) self.assertIsInstance(bounds_rect, pygame.Rect) self.assertEqual(bounds_rect, pygame.Rect(0, 0, 2, 2)) def test_circle__kwargs(self): """Ensures draw circle accepts the correct kwargs with and without a width and quadrant arguments. """ kwargs_list = [ { "surface": pygame.Surface((4, 4)), "color": pygame.Color("yellow"), "center": (2, 2), "radius": 2, "width": 1, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": False, "draw_bottom_right": True, }, { "surface": pygame.Surface((2, 1)), "color": (0, 10, 20), "center": (1, 1), "radius": 1, }, ] for kwargs in kwargs_list: bounds_rect = self.draw_circle(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__kwargs_order_independent(self): """Ensures draw circle's kwargs are not order dependent.""" bounds_rect = self.draw_circle( draw_top_right=False, color=(10, 20, 30), surface=pygame.Surface((3, 2)), width=0, draw_bottom_left=False, center=(1, 0), draw_bottom_right=False, radius=2, draw_top_left=True, ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__args_missing(self): """Ensures draw circle detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("blue") with self.assertRaises(TypeError): bounds_rect = self.draw_circle(surface, color, (0, 0)) with self.assertRaises(TypeError): bounds_rect = self.draw_circle(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_circle(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_circle() def test_circle__kwargs_missing(self): """Ensures draw circle detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((1, 2)), "color": pygame.Color("red"), "center": (1, 0), "radius": 2, "width": 1, "draw_top_right": False, "draw_top_left": False, "draw_bottom_left": False, "draw_bottom_right": True, } for name in ("radius", "center", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_circle(**invalid_kwargs) def test_circle__arg_invalid_types(self): """Ensures draw circle detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") center = (1, 1) radius = 1 with self.assertRaises(TypeError): # Invalid draw_top_right. bounds_rect = self.draw_circle( surface, color, center, radius, 1, "a", 1, 1, 1 ) with self.assertRaises(TypeError): # Invalid draw_top_left. bounds_rect = self.draw_circle( surface, color, center, radius, 1, 1, "b", 1, 1 ) with self.assertRaises(TypeError): # Invalid draw_bottom_left. bounds_rect = self.draw_circle( surface, color, center, radius, 1, 1, 1, "c", 1 ) with self.assertRaises(TypeError): # Invalid draw_bottom_right. bounds_rect = self.draw_circle( surface, color, center, radius, 1, 1, 1, 1, "d" ) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_circle(surface, color, center, radius, "1") with self.assertRaises(TypeError): # Invalid radius. bounds_rect = self.draw_circle(surface, color, center, "2") with self.assertRaises(TypeError): # Invalid center. bounds_rect = self.draw_circle(surface, color, (1, 2, 3), radius) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_circle(surface, 2.3, center, radius) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_circle((1, 2, 3, 4), color, center, radius) def test_circle__kwarg_invalid_types(self): """Ensures draw circle detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") center = (0, 1) radius = 1 width = 1 quadrant = 1 kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": 2.3, # Invalid color. "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": (1, 1, 1), # Invalid center. "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": "1", # Invalid radius. "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "width": 1.2, # Invalid width. "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": "True", # Invalid draw_top_right "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": "True", # Invalid draw_top_left "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": 3.14, # Invalid draw_bottom_left "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": "quadrant", # Invalid draw_bottom_right }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_circle(**kwargs) def test_circle__kwarg_invalid_name(self): """Ensures draw circle detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") center = (0, 0) radius = 2 kwargs_list = [ { "surface": surface, "color": color, "center": center, "radius": radius, "width": 1, "quadrant": 1, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, }, { "surface": surface, "color": color, "center": center, "radius": radius, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_circle(**kwargs) def test_circle__args_and_kwargs(self): """Ensures draw circle accepts a combination of args/kwargs""" surface = pygame.Surface((3, 1)) color = (255, 255, 0, 0) center = (1, 0) radius = 2 width = 0 draw_top_right = True draw_top_left = False draw_bottom_left = False draw_bottom_right = True kwargs = { "surface": surface, "color": color, "center": center, "radius": radius, "width": width, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } for name in ( "surface", "color", "center", "radius", "width", "draw_top_right", "draw_top_left", "draw_bottom_left", "draw_bottom_right", ): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_circle(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_circle(surface, color, **kwargs) elif "center" == name: bounds_rect = self.draw_circle(surface, color, center, **kwargs) elif "radius" == name: bounds_rect = self.draw_circle(surface, color, center, radius, **kwargs) elif "width" == name: bounds_rect = self.draw_circle( surface, color, center, radius, width, **kwargs ) elif "draw_top_right" == name: bounds_rect = self.draw_circle( surface, color, center, radius, width, draw_top_right, **kwargs ) elif "draw_top_left" == name: bounds_rect = self.draw_circle( surface, color, center, radius, width, draw_top_right, draw_top_left, **kwargs, ) elif "draw_bottom_left" == name: bounds_rect = self.draw_circle( surface, color, center, radius, width, draw_top_right, draw_top_left, draw_bottom_left, **kwargs, ) else: bounds_rect = self.draw_circle( surface, color, center, radius, width, draw_top_right, draw_top_left, draw_bottom_left, draw_bottom_right, **kwargs, ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__valid_width_values(self): """Ensures draw circle accepts different width values.""" center = (2, 2) radius = 1 pos = (center[0] - radius, center[1]) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) color = (10, 20, 30, 255) kwargs = { "surface": surface, "color": color, "center": center, "radius": radius, "width": None, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } for width in (-100, -10, -1, 0, 1, 10, 100): surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = color if width >= 0 else surface_color bounds_rect = self.draw_circle(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__valid_radius_values(self): """Ensures draw circle accepts different radius values.""" pos = center = (2, 2) surface_color = pygame.Color("white") surface = pygame.Surface((3, 4)) color = (10, 20, 30, 255) kwargs = { "surface": surface, "color": color, "center": center, "radius": None, "width": 0, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } for radius in (-10, -1, 0, 1, 10): surface.fill(surface_color) # Clear for each test. kwargs["radius"] = radius expected_color = color if radius > 0 else surface_color bounds_rect = self.draw_circle(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__valid_center_formats(self): """Ensures draw circle accepts different center formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((4, 4)) kwargs = { "surface": surface, "color": expected_color, "center": None, "radius": 1, "width": 0, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } x, y = 2, 2 # center position # The center values can be ints or floats. for center in ((x, y), (x + 0.1, y), (x, y + 0.1), (x + 0.1, y + 0.1)): # The center type can be a tuple/list/Vector2. for seq_type in (tuple, list, Vector2): surface.fill(surface_color) # Clear for each test. kwargs["center"] = seq_type(center) bounds_rect = self.draw_circle(**kwargs) self.assertEqual(surface.get_at((x, y)), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__valid_color_formats(self): """Ensures draw circle accepts different color formats.""" center = (2, 2) radius = 1 pos = (center[0] - radius, center[1]) green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((3, 4)) kwargs = { "surface": surface, "color": None, "center": center, "radius": radius, "width": 0, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_circle(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_circle__invalid_color_formats(self): """Ensures draw circle handles invalid color formats correctly.""" kwargs = { "surface": pygame.Surface((4, 3)), "color": None, "center": (1, 2), "radius": 1, "width": 0, "draw_top_right": True, "draw_top_left": True, "draw_bottom_left": True, "draw_bottom_right": True, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_circle(**kwargs) def test_circle__floats(self): """Ensure that floats are accepted.""" draw.circle( surface=pygame.Surface((4, 4)), color=(255, 255, 127), center=(1.5, 1.5), radius=1.3, width=0, draw_top_right=True, draw_top_left=True, draw_bottom_left=True, draw_bottom_right=True, ) draw.circle( surface=pygame.Surface((4, 4)), color=(255, 255, 127), center=Vector2(1.5, 1.5), radius=1.3, width=0, draw_top_right=True, draw_top_left=True, draw_bottom_left=True, draw_bottom_right=True, ) draw.circle(pygame.Surface((2, 2)), (0, 0, 0, 50), (1.3, 1.3), 1.2) # def test_circle_clip(self): # """ maybe useful to help work out circle clip algorithm.""" # MAX = max # MIN = min # posx=30 # posy=15 # radius=1 # l=29 # t=14 # r=30 # b=16 # clip_rect_x=0 # clip_rect_y=0 # clip_rect_w=30 # clip_rect_h=30 # l = MAX(posx - radius, clip_rect_x) # t = MAX(posy - radius, clip_rect_y) # r = MIN(posx + radius, clip_rect_x + clip_rect_w) # b = MIN(posy + radius, clip_rect_y + clip_rect_h) # l, t, MAX(r - l, 0), MAX(b - t, 0) def test_circle__bounding_rect(self): """Ensures draw circle returns the correct bounding rect. Tests circles on and off the surface and a range of width/thickness values. """ circle_color = pygame.Color("red") surf_color = pygame.Color("black") max_radius = 3 surface = pygame.Surface((30, 30), 0, 32) surf_rect = surface.get_rect() # Make a rect that is bigger than the surface to help test drawing # circles off and partially off the surface. Make this rect such that # when centering the test circle on one of its corners, the circle is # drawn fully off the test surface, but a rect bounding the circle # would still overlap with the test surface. big_rect = surf_rect.inflate(max_radius * 2 - 1, max_radius * 2 - 1) for pos in rect_corners_mids_and_center( surf_rect ) + rect_corners_mids_and_center(big_rect): # Test using different radius and thickness values. for radius in range(max_radius + 1): for thickness in range(radius + 1): surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_circle( surface, circle_color, pos, radius, thickness ) # Calculating the expected_rect after the circle is # drawn (it uses what is actually drawn). expected_rect = create_bounding_rect(surface, surf_color, pos) # print("pos:%s:, radius:%s:, thickness:%s:" % (pos, radius, thickness)) self.assertEqual(bounding_rect, expected_rect) def test_circle_negative_radius(self): """Ensures negative radius circles return zero sized bounding rect.""" surf = pygame.Surface((200, 200)) color = (0, 0, 0, 50) center = surf.get_height() // 2, surf.get_height() // 2 bounding_rect = self.draw_circle(surf, color, center, radius=-1, width=1) self.assertEqual(bounding_rect.size, (0, 0)) def test_circle_zero_radius(self): """Ensures zero radius circles does not draw a center pixel. NOTE: This is backwards incompatible behaviour with 1.9.x. """ surf = pygame.Surface((200, 200)) circle_color = pygame.Color("red") surf_color = pygame.Color("black") center = (100, 100) radius = 0 width = 1 bounding_rect = self.draw_circle(surf, circle_color, center, radius, width) expected_rect = create_bounding_rect(surf, surf_color, center) self.assertEqual(bounding_rect, expected_rect) self.assertEqual(bounding_rect, pygame.Rect(100, 100, 0, 0)) def test_circle__surface_clip(self): """Ensures draw circle respects a surface's clip area. Tests drawing the circle filled and unfilled. """ surfw = surfh = 25 circle_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (10, 10)) clip_rect.center = surface.get_rect().center radius = clip_rect.w // 2 + 1 for width in (0, 1): # Filled and unfilled. # Test centering the circle along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the circle without the # clip area set. surface.set_clip(None) surface.fill(surface_color) self.draw_circle(surface, circle_color, center, radius, width) expected_pts = get_color_points(surface, circle_color, clip_rect) # Clear the surface and set the clip area. Redraw the circle # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_circle(surface, circle_color, center, radius, width) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the circle_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = circle_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() def test_circle_shape(self): """Ensures there are no holes in the circle, and no overdrawing. Tests drawing a thick circle. Measures the distance of the drawn pixels from the circle center. """ surfw = surfh = 100 circle_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) (cx, cy) = center = (50, 50) radius = 45 width = 25 dest_rect = self.draw_circle(surface, circle_color, center, radius, width) for pt in test_utils.rect_area_pts(dest_rect): x, y = pt sqr_distance = (x - cx) ** 2 + (y - cy) ** 2 if (radius - width + 1) ** 2 < sqr_distance < (radius - 1) ** 2: self.assertEqual(surface.get_at(pt), circle_color) if ( sqr_distance < (radius - width - 1) ** 2 or sqr_distance > (radius + 1) ** 2 ): self.assertEqual(surface.get_at(pt), surface_color) def test_circle__diameter(self): """Ensures draw circle is twice size of radius high and wide.""" surf = pygame.Surface((200, 200)) color = (0, 0, 0, 50) center = surf.get_height() // 2, surf.get_height() // 2 width = 1 radius = 6 for radius in range(1, 65): bounding_rect = self.draw_circle(surf, color, center, radius, width) self.assertEqual(bounding_rect.width, radius * 2) self.assertEqual(bounding_rect.height, radius * 2) class DrawCircleTest(DrawCircleMixin, DrawTestCase): """Test draw module function circle. This class inherits the general tests from DrawCircleMixin. It is also the class to add any draw.circle specific tests to. """ # Commented out to avoid cluttering the test output. Add back in if draw_py # ever properly supports drawing circles. # @unittest.skip('draw_py.draw_circle not supported yet') # class PythonDrawCircleTest(DrawCircleMixin, PythonDrawTestCase): # """Test draw_py module function draw_circle." # # This class inherits the general tests from DrawCircleMixin. It is also # the class to add any draw_py.draw_circle specific tests to. # """ ### Arc Testing ############################################################### class DrawArcMixin: """Mixin tests for drawing arcs. This class contains all the general arc drawing tests. """ def test_arc__args(self): """Ensures draw arc accepts the correct args.""" bounds_rect = self.draw_arc( pygame.Surface((3, 3)), (0, 10, 0, 50), (1, 1, 2, 2), 0, 1, 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__args_without_width(self): """Ensures draw arc accepts the args without a width.""" bounds_rect = self.draw_arc( pygame.Surface((2, 2)), (1, 1, 1, 99), pygame.Rect((0, 0), (2, 2)), 1.1, 2.1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__args_with_negative_width(self): """Ensures draw arc accepts the args with negative width.""" bounds_rect = self.draw_arc( pygame.Surface((3, 3)), (10, 10, 50, 50), (1, 1, 2, 2), 0, 1, -1 ) self.assertIsInstance(bounds_rect, pygame.Rect) self.assertEqual(bounds_rect, pygame.Rect(1, 1, 0, 0)) def test_arc__args_with_width_gt_radius(self): """Ensures draw arc accepts the args with width > rect.w // 2 and width > rect.h // 2. """ rect = pygame.Rect((0, 0), (4, 4)) bounds_rect = self.draw_arc( pygame.Surface((3, 3)), (10, 10, 50, 50), rect, 0, 45, rect.w // 2 + 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) bounds_rect = self.draw_arc( pygame.Surface((3, 3)), (10, 10, 50, 50), rect, 0, 45, rect.h // 2 + 1 ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__kwargs(self): """Ensures draw arc accepts the correct kwargs with and without a width arg. """ kwargs_list = [ { "surface": pygame.Surface((4, 4)), "color": pygame.Color("yellow"), "rect": pygame.Rect((0, 0), (3, 2)), "start_angle": 0.5, "stop_angle": 3, "width": 1, }, { "surface": pygame.Surface((2, 1)), "color": (0, 10, 20), "rect": (0, 0, 2, 2), "start_angle": 1, "stop_angle": 3.1, }, ] for kwargs in kwargs_list: bounds_rect = self.draw_arc(**kwargs) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__kwargs_order_independent(self): """Ensures draw arc's kwargs are not order dependent.""" bounds_rect = self.draw_arc( stop_angle=1, start_angle=2.2, color=(1, 2, 3), surface=pygame.Surface((3, 2)), width=1, rect=pygame.Rect((1, 0), (2, 3)), ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__args_missing(self): """Ensures draw arc detects any missing required args.""" surface = pygame.Surface((1, 1)) color = pygame.Color("red") rect = pygame.Rect((0, 0), (2, 2)) with self.assertRaises(TypeError): bounds_rect = self.draw_arc(surface, color, rect, 0.1) with self.assertRaises(TypeError): bounds_rect = self.draw_arc(surface, color, rect) with self.assertRaises(TypeError): bounds_rect = self.draw_arc(surface, color) with self.assertRaises(TypeError): bounds_rect = self.draw_arc(surface) with self.assertRaises(TypeError): bounds_rect = self.draw_arc() def test_arc__kwargs_missing(self): """Ensures draw arc detects any missing required kwargs.""" kwargs = { "surface": pygame.Surface((1, 2)), "color": pygame.Color("red"), "rect": pygame.Rect((1, 0), (2, 2)), "start_angle": 0.1, "stop_angle": 2, "width": 1, } for name in ("stop_angle", "start_angle", "rect", "color", "surface"): invalid_kwargs = dict(kwargs) invalid_kwargs.pop(name) # Pop from a copy. with self.assertRaises(TypeError): bounds_rect = self.draw_arc(**invalid_kwargs) def test_arc__arg_invalid_types(self): """Ensures draw arc detects invalid arg types.""" surface = pygame.Surface((2, 2)) color = pygame.Color("blue") rect = pygame.Rect((1, 1), (3, 3)) with self.assertRaises(TypeError): # Invalid width. bounds_rect = self.draw_arc(surface, color, rect, 0, 1, "1") with self.assertRaises(TypeError): # Invalid stop_angle. bounds_rect = self.draw_arc(surface, color, rect, 0, "1", 1) with self.assertRaises(TypeError): # Invalid start_angle. bounds_rect = self.draw_arc(surface, color, rect, "1", 0, 1) with self.assertRaises(TypeError): # Invalid rect. bounds_rect = self.draw_arc(surface, color, (1, 2, 3, 4, 5), 0, 1, 1) with self.assertRaises(TypeError): # Invalid color. bounds_rect = self.draw_arc(surface, 2.3, rect, 0, 1, 1) with self.assertRaises(TypeError): # Invalid surface. bounds_rect = self.draw_arc(rect, color, rect, 0, 1, 1) def test_arc__kwarg_invalid_types(self): """Ensures draw arc detects invalid kwarg types.""" surface = pygame.Surface((3, 3)) color = pygame.Color("green") rect = pygame.Rect((0, 1), (4, 2)) start = 3 stop = 4 kwargs_list = [ { "surface": pygame.Surface, # Invalid surface. "color": color, "rect": rect, "start_angle": start, "stop_angle": stop, "width": 1, }, { "surface": surface, "color": 2.3, # Invalid color. "rect": rect, "start_angle": start, "stop_angle": stop, "width": 1, }, { "surface": surface, "color": color, "rect": (0, 0, 0), # Invalid rect. "start_angle": start, "stop_angle": stop, "width": 1, }, { "surface": surface, "color": color, "rect": rect, "start_angle": "1", # Invalid start_angle. "stop_angle": stop, "width": 1, }, { "surface": surface, "color": color, "rect": rect, "start_angle": start, "stop_angle": "1", # Invalid stop_angle. "width": 1, }, { "surface": surface, "color": color, "rect": rect, "start_angle": start, "stop_angle": stop, "width": 1.1, }, ] # Invalid width. for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_arc(**kwargs) def test_arc__kwarg_invalid_name(self): """Ensures draw arc detects invalid kwarg names.""" surface = pygame.Surface((2, 3)) color = pygame.Color("cyan") rect = pygame.Rect((0, 1), (2, 2)) start = 0.9 stop = 2.3 kwargs_list = [ { "surface": surface, "color": color, "rect": rect, "start_angle": start, "stop_angle": stop, "width": 1, "invalid": 1, }, { "surface": surface, "color": color, "rect": rect, "start_angle": start, "stop_angle": stop, "invalid": 1, }, ] for kwargs in kwargs_list: with self.assertRaises(TypeError): bounds_rect = self.draw_arc(**kwargs) def test_arc__args_and_kwargs(self): """Ensures draw arc accepts a combination of args/kwargs""" surface = pygame.Surface((3, 1)) color = (255, 255, 0, 0) rect = pygame.Rect((1, 0), (2, 3)) start = 0.6 stop = 2 width = 1 kwargs = { "surface": surface, "color": color, "rect": rect, "start_angle": start, "stop_angle": stop, "width": width, } for name in ("surface", "color", "rect", "start_angle", "stop_angle"): kwargs.pop(name) if "surface" == name: bounds_rect = self.draw_arc(surface, **kwargs) elif "color" == name: bounds_rect = self.draw_arc(surface, color, **kwargs) elif "rect" == name: bounds_rect = self.draw_arc(surface, color, rect, **kwargs) elif "start_angle" == name: bounds_rect = self.draw_arc(surface, color, rect, start, **kwargs) elif "stop_angle" == name: bounds_rect = self.draw_arc(surface, color, rect, start, stop, **kwargs) else: bounds_rect = self.draw_arc( surface, color, rect, start, stop, width, **kwargs ) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__valid_width_values(self): """Ensures draw arc accepts different width values.""" arc_color = pygame.Color("yellow") surface_color = pygame.Color("white") surface = pygame.Surface((6, 6)) rect = pygame.Rect((0, 0), (4, 4)) rect.center = surface.get_rect().center pos = rect.centerx + 1, rect.centery + 1 kwargs = { "surface": surface, "color": arc_color, "rect": rect, "start_angle": 0, "stop_angle": 7, "width": None, } for width in (-50, -10, -3, -2, -1, 0, 1, 2, 3, 10, 50): msg = f"width={width}" surface.fill(surface_color) # Clear for each test. kwargs["width"] = width expected_color = arc_color if width > 0 else surface_color bounds_rect = self.draw_arc(**kwargs) self.assertEqual(surface.get_at(pos), expected_color, msg) self.assertIsInstance(bounds_rect, pygame.Rect, msg) def test_arc__valid_stop_angle_values(self): """Ensures draw arc accepts different stop_angle values.""" expected_color = pygame.Color("blue") surface_color = pygame.Color("white") surface = pygame.Surface((6, 6)) rect = pygame.Rect((0, 0), (4, 4)) rect.center = surface.get_rect().center pos = rect.centerx, rect.centery + 1 kwargs = { "surface": surface, "color": expected_color, "rect": rect, "start_angle": -17, "stop_angle": None, "width": 1, } for stop_angle in (-10, -5.5, -1, 0, 1, 5.5, 10): msg = f"stop_angle={stop_angle}" surface.fill(surface_color) # Clear for each test. kwargs["stop_angle"] = stop_angle bounds_rect = self.draw_arc(**kwargs) self.assertEqual(surface.get_at(pos), expected_color, msg) self.assertIsInstance(bounds_rect, pygame.Rect, msg) def test_arc__valid_start_angle_values(self): """Ensures draw arc accepts different start_angle values.""" expected_color = pygame.Color("blue") surface_color = pygame.Color("white") surface = pygame.Surface((6, 6)) rect = pygame.Rect((0, 0), (4, 4)) rect.center = surface.get_rect().center pos = rect.centerx + 1, rect.centery + 1 kwargs = { "surface": surface, "color": expected_color, "rect": rect, "start_angle": None, "stop_angle": 17, "width": 1, } for start_angle in (-10.0, -5.5, -1, 0, 1, 5.5, 10.0): msg = f"start_angle={start_angle}" surface.fill(surface_color) # Clear for each test. kwargs["start_angle"] = start_angle bounds_rect = self.draw_arc(**kwargs) self.assertEqual(surface.get_at(pos), expected_color, msg) self.assertIsInstance(bounds_rect, pygame.Rect, msg) def test_arc__valid_rect_formats(self): """Ensures draw arc accepts different rect formats.""" expected_color = pygame.Color("red") surface_color = pygame.Color("black") surface = pygame.Surface((6, 6)) rect = pygame.Rect((0, 0), (4, 4)) rect.center = surface.get_rect().center pos = rect.centerx + 1, rect.centery + 1 kwargs = { "surface": surface, "color": expected_color, "rect": None, "start_angle": 0, "stop_angle": 7, "width": 1, } rects = (rect, (rect.topleft, rect.size), (rect.x, rect.y, rect.w, rect.h)) for rect in rects: surface.fill(surface_color) # Clear for each test. kwargs["rect"] = rect bounds_rect = self.draw_arc(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__valid_color_formats(self): """Ensures draw arc accepts different color formats.""" green_color = pygame.Color("green") surface_color = pygame.Color("black") surface = pygame.Surface((6, 6)) rect = pygame.Rect((0, 0), (4, 4)) rect.center = surface.get_rect().center pos = rect.centerx + 1, rect.centery + 1 kwargs = { "surface": surface, "color": None, "rect": rect, "start_angle": 0, "stop_angle": 7, "width": 1, } greens = ( (0, 255, 0), (0, 255, 0, 255), surface.map_rgb(green_color), green_color, ) for color in greens: surface.fill(surface_color) # Clear for each test. kwargs["color"] = color if isinstance(color, int): expected_color = surface.unmap_rgb(color) else: expected_color = green_color bounds_rect = self.draw_arc(**kwargs) self.assertEqual(surface.get_at(pos), expected_color) self.assertIsInstance(bounds_rect, pygame.Rect) def test_arc__invalid_color_formats(self): """Ensures draw arc handles invalid color formats correctly.""" pos = (1, 1) surface = pygame.Surface((4, 3)) kwargs = { "surface": surface, "color": None, "rect": pygame.Rect(pos, (2, 2)), "start_angle": 5, "stop_angle": 6.1, "width": 1, } for expected_color in (2.3, self): kwargs["color"] = expected_color with self.assertRaises(TypeError): bounds_rect = self.draw_arc(**kwargs) def test_arc(self): """Ensure draw arc works correctly.""" black = pygame.Color("black") red = pygame.Color("red") # create an image object of width 100, height 150, filled with black. surface = pygame.Surface((100, 150)) surface.fill(black) # rectangle that contains for the ellipse arc. # 0 pixel from left, 0 pixel from top # 80 pixels wide, 40 pixels high rect = (0, 0, 80, 40) # angle of the arc in radians start_angle = 0.0 stop_angle = 3.14 # thickness, and it grows inward from the rectangle width = 3 # draw an elliptical arc pygame.draw.arc(surface, red, rect, start_angle, stop_angle, width) # Save the drawn arc pygame.image.save(surface, "arc.png") # arc is red x = 20 for y in range(2, 5): self.assertEqual(surface.get_at((x, y)), red) # the rest area in surface is black self.assertEqual(surface.get_at((0, 0)), black) def test_arc__bounding_rect(self): """Ensures draw arc returns the correct bounding rect. Tests arcs on and off the surface and a range of width/thickness values. """ arc_color = pygame.Color("red") surf_color = pygame.Color("black") min_width = min_height = 5 max_width = max_height = 7 sizes = ((min_width, min_height), (max_width, max_height)) surface = pygame.Surface((20, 20), 0, 32) surf_rect = surface.get_rect() # Make a rect that is bigger than the surface to help test drawing # arcs off and partially off the surface. big_rect = surf_rect.inflate(min_width * 2 + 1, min_height * 2 + 1) # Max angle allows for a full circle to be drawn. start_angle = 0 stop_angles = (0, 2, 3, 5, math.ceil(2 * math.pi)) for pos in rect_corners_mids_and_center( surf_rect ) + rect_corners_mids_and_center(big_rect): # Each of the arc's rect position attributes will be set to the pos # value. for attr in RECT_POSITION_ATTRIBUTES: # Test using different rect sizes, thickness values and stop # angles. for width, height in sizes: arc_rect = pygame.Rect((0, 0), (width, height)) setattr(arc_rect, attr, pos) for thickness in (0, 1, 2, 3, min(width, height)): for stop_angle in stop_angles: surface.fill(surf_color) # Clear for each test. bounding_rect = self.draw_arc( surface, arc_color, arc_rect, start_angle, stop_angle, thickness, ) # Calculating the expected_rect after the arc # is drawn (it uses what is actually drawn). expected_rect = create_bounding_rect( surface, surf_color, arc_rect.topleft ) self.assertEqual( bounding_rect, expected_rect, f"thickness={thickness}", ) def test_arc__surface_clip(self): """Ensures draw arc respects a surface's clip area.""" surfw = surfh = 30 start = 0.1 end = 0 # end < start so a full circle will be drawn arc_color = pygame.Color("red") surface_color = pygame.Color("green") surface = pygame.Surface((surfw, surfh)) surface.fill(surface_color) clip_rect = pygame.Rect((0, 0), (11, 11)) clip_rect.center = surface.get_rect().center pos_rect = clip_rect.copy() # Manages the arc's pos. for thickness in (1, 3): # Different line widths. # Test centering the arc along the clip rect's edge. for center in rect_corners_mids_and_center(clip_rect): # Get the expected points by drawing the arc without the # clip area set. pos_rect.center = center surface.set_clip(None) surface.fill(surface_color) self.draw_arc(surface, arc_color, pos_rect, start, end, thickness) expected_pts = get_color_points(surface, arc_color, clip_rect) # Clear the surface and set the clip area. Redraw the arc # and check that only the clip area is modified. surface.fill(surface_color) surface.set_clip(clip_rect) self.draw_arc(surface, arc_color, pos_rect, start, end, thickness) surface.lock() # For possible speed up. # Check all the surface points to ensure only the expected_pts # are the arc_color. for pt in ((x, y) for x in range(surfw) for y in range(surfh)): if pt in expected_pts: expected_color = arc_color else: expected_color = surface_color self.assertEqual(surface.get_at(pt), expected_color, pt) surface.unlock() class DrawArcTest(DrawArcMixin, DrawTestCase): """Test draw module function arc. This class inherits the general tests from DrawArcMixin. It is also the class to add any draw.arc specific tests to. """ # Commented out to avoid cluttering the test output. Add back in if draw_py # ever properly supports drawing arcs. # @unittest.skip('draw_py.draw_arc not supported yet') # class PythonDrawArcTest(DrawArcMixin, PythonDrawTestCase): # """Test draw_py module function draw_arc. # # This class inherits the general tests from DrawArcMixin. It is also the # class to add any draw_py.draw_arc specific tests to. # """ ### Draw Module Testing ####################################################### class DrawModuleTest(unittest.TestCase): """General draw module tests.""" def test_path_data_validation(self): """Test validation of multi-point drawing methods. See bug #521 """ surf = pygame.Surface((5, 5)) rect = pygame.Rect(0, 0, 5, 5) bad_values = ( "text", b"bytes", 1 + 1j, # string, bytes, complex, object(), (lambda x: x), ) # object, function bad_points = list(bad_values) + [(1,), (1, 2, 3)] # wrong tuple length bad_points.extend((1, v) for v in bad_values) # one wrong value good_path = [(1, 1), (1, 3), (3, 3), (3, 1)] # A) draw.lines check_pts = [(x, y) for x in range(5) for y in range(5)] for method, is_polgon in ( (draw.lines, 0), (draw.aalines, 0), (draw.polygon, 1), ): for val in bad_values: # 1. at the beginning draw.rect(surf, RED, rect, 0) with self.assertRaises(TypeError): if is_polgon: method(surf, GREEN, [val] + good_path, 0) else: method(surf, GREEN, True, [val] + good_path) # make sure, nothing was drawn : self.assertTrue(all(surf.get_at(pt) == RED for pt in check_pts)) # 2. not at the beginning (was not checked) draw.rect(surf, RED, rect, 0) with self.assertRaises(TypeError): path = good_path[:2] + [val] + good_path[2:] if is_polgon: method(surf, GREEN, path, 0) else: method(surf, GREEN, True, path) # make sure, nothing was drawn : self.assertTrue(all(surf.get_at(pt) == RED for pt in check_pts)) def test_color_validation(self): surf = pygame.Surface((10, 10)) colors = 123456, (1, 10, 100), RED, "#ab12df", "red" points = ((0, 0), (1, 1), (1, 0)) # 1. valid colors for col in colors: draw.line(surf, col, (0, 0), (1, 1)) draw.aaline(surf, col, (0, 0), (1, 1)) draw.aalines(surf, col, True, points) draw.lines(surf, col, True, points) draw.arc(surf, col, pygame.Rect(0, 0, 3, 3), 15, 150) draw.ellipse(surf, col, pygame.Rect(0, 0, 3, 6), 1) draw.circle(surf, col, (7, 3), 2) draw.polygon(surf, col, points, 0) # 2. invalid colors for col in (1.256, object(), None): with self.assertRaises(TypeError): draw.line(surf, col, (0, 0), (1, 1)) with self.assertRaises(TypeError): draw.aaline(surf, col, (0, 0), (1, 1)) with self.assertRaises(TypeError): draw.aalines(surf, col, True, points) with self.assertRaises(TypeError): draw.lines(surf, col, True, points) with self.assertRaises(TypeError): draw.arc(surf, col, pygame.Rect(0, 0, 3, 3), 15, 150) with self.assertRaises(TypeError): draw.ellipse(surf, col, pygame.Rect(0, 0, 3, 6), 1) with self.assertRaises(TypeError): draw.circle(surf, col, (7, 3), 2) with self.assertRaises(TypeError): draw.polygon(surf, col, points, 0) ############################################################################### if __name__ == "__main__": unittest.main()