def test_scale():
surface = NumpySurface(width=200, height=100, fill=[0, 0, 0])
red_box = NumpySurface(width=100, height=50, fill=[255, 0, 0])
surface.blit(red_box, location=(0, 0), blend=False)
surf = surface.copy()
surf.scale(width=300)
assert surf.height == 150
assert surf.content[0][149][0] > 128 and surf.content[74][0][0] > 128
assert surf.content[0][150][0] < 64 and surf.content[75][0][0] < 64
surf = surface.copy()
surf.scale(height=50)
assert surf.width == 100
assert surf.content[0][49][0] > 128 and surf.content[24][0][0] > 128
assert surf.content[0][50][0] < 64 and surf.content[25][0][0] < 64
surf = surface.copy()
surf.scale(width=500, height=500)
assert surf.height == 500 and surf.width == 500
assert surf.content[0][249][0] > 128 and surf.content[249][0][0] > 128
assert surf.content[0][250][0] < 128 and surf.content[250][0][0] < 128
with pytest.raises(ValueError):
surf.scale()
def test_blit():
# Initialize different test surfaces
surf_clear = NumpySurface(width=100, height=100, fill=[0, 0, 0, 0])
surf_black = NumpySurface(width=100, height=100, fill=[0, 0, 0, 255])
test_surf = maketestsurface()
test_square = NumpySurface(width=50, height=50, fill=(255, 255, 255))
# Test registrations
expected_bounds = {
'1': (50, 0, 100, 50),
'2': (25, 0, 75, 50),
'3': (0, 0, 50, 50),
'4': (50, 25, 100, 75),
'5': (25, 25, 75, 75),
'6': (0, 25, 50, 75),
'7': (50, 50, 100, 100),
'8': (25, 50, 75, 100),
'9': (0, 50, 50, 100),
}
for r in [1, 2, 3, 4, 5, 6, 7, 8, 9]:
for blendmode in [True, False]:
surf = surf_clear.copy()
sc = surf.surface_c
surf.blit(test_square,
registration=r,
location=sc,
blend=blendmode)
img = Image.fromarray(surf.content)
assert expected_bounds[str(r)] == img.getbbox()
# Test blending behaviour
surf = surf_black.copy()
surf.blit(test_surf, registration=7, location=(0, 0), blend=True)
assert surf.content[0][0][3] == 255 and surf.content[-1][-1][3] == 255
surf = surf_black.copy()
surf.blit(test_surf, registration=7, location=(0, 0), blend=False)
assert surf.content[0][0][3] == 255 and surf.content[-1][-1][3] == 0
# Test clipping behaviour
for blendmode in [True, False]:
surf = surf_clear.copy()
surf.blit(test_square, location=(75, 75), blend=blendmode, clip=True)
surf.blit(test_square, location=(-25, -25), blend=blendmode, clip=True)
assert surf.content[24][24][0] == 255 and surf.content[25][25][0] == 0
assert surf.content[75][75][0] == 255 and surf.content[74][74][0] == 0
# Test blit exceptions
with pytest.raises(TypeError):
surf = surf_clear.copy()
surf.blit(1, location=(0, 0))
with pytest.raises(ValueError):
surf = surf_clear.copy()
surf.blit(test_square, location=(150, 150))
with pytest.raises(ValueError):
surf = surf_clear.copy()
surf.blit(test_square, location=(75, 75), clip=False)
with pytest.raises(ValueError):
surf = test_square.copy()
surf.blit(surf_clear, location=(0, 0), clip=False)
def test_copy():
surf = NumpySurface(width=100, height=100, fill=[0, 0, 0])
surf_copy = surf.copy()
assert isinstance(surf_copy, NumpySurface)
surf.content[0][0][0] = 255
assert surf.content[0][0][0] != surf_copy.content[0][0][0]
def test_mask():
# Initialize test surface and masks
# NOTE: Testing KLDraw objects further down, since KLD rectangles have 1px
# transparent padding that breaks this test loop's logic
surface = NumpySurface(width=100, height=100, fill=[255, 0, 0])
nps_mask = NumpySurface(width=50, height=50, fill=[255, 255, 255])
np_mask = nps_mask.render()
greyscale_mask = Image.new('L', (50, 50), 0)
ImageDraw.Draw(greyscale_mask).rectangle((0, 0, 50, 50), fill=255)
rgb_mask = Image.new('RGB', (50, 50), (0, 0, 0))
ImageDraw.Draw(rgb_mask).rectangle((0, 0, 50, 50), fill=(255, 0, 0))
# Test different mask types
for mask in [nps_mask, np_mask, greyscale_mask, rgb_mask]:
surf = surface.copy()
surf.mask(mask, registration=7, location=(0, 0))
assert surf.content[49][49][3] == 0
assert surf.content[50][50][3] == 255
# Test legacy positioning
surf = surface.copy()
surf.mask(nps_mask, (25, 25))
assert surf.content[0][0][3] == 255 and surf.content[25][25][3] == 0
# Test with mask partially off surface
surf = surface.copy()
surf.mask(nps_mask, registration=3, location=(25, 25))
assert surf.content[24][24][3] == 0 and surf.content[25][25][3] == 255
surf.mask(nps_mask, registration=7, location=(75, 75))
assert surf.content[74][74][3] == 255 and surf.content[75][75][3] == 0
# Test inverse/non-inverse modes and complete masking
circle_mask = kld.Ellipse(50, fill=(255, 255, 255))
for complete in [True, False]:
surf = surface.copy()
surf.mask(circle_mask, location=(0, 0), invert=True, complete=complete)
assert surf.content[0][0][3] == 255 and surf.content[25][25][3] == 0
assert surf.content[-1][-1][3] == (0 if complete else 255)
surf = surface.copy()
surf.mask(circle_mask,
location=(0, 0),
invert=False,
complete=complete)
assert surf.content[0][0][3] == 0 and surf.content[25][25][3] == 255
assert surf.content[-1][-1][3] == (0 if complete else 255)
# Test exception for invalid mask type
with pytest.raises(TypeError):
surf = surface.copy()
surf.mask("hello")