790 lines
28 KiB
Python
790 lines
28 KiB
Python
if __name__ == '__main__':
|
|
import sys
|
|
import os
|
|
pkg_dir = os.path.split(os.path.abspath(__file__))[0]
|
|
parent_dir, pkg_name = os.path.split(pkg_dir)
|
|
is_pygame_pkg = (pkg_name == 'tests' and
|
|
os.path.split(parent_dir)[1] == 'pygame')
|
|
if not is_pygame_pkg:
|
|
sys.path.insert(0, parent_dir)
|
|
else:
|
|
is_pygame_pkg = __name__.startswith('pygame.tests.')
|
|
|
|
import unittest
|
|
from pygame import Rect
|
|
|
|
class RectTypeTest( unittest.TestCase ):
|
|
def testConstructionXYWidthHeight( self ):
|
|
r = Rect(1,2,3,4)
|
|
self.assertEqual( 1, r.left )
|
|
self.assertEqual( 2, r.top )
|
|
self.assertEqual( 3, r.width )
|
|
self.assertEqual( 4, r.height )
|
|
|
|
def testConstructionTopLeftSize( self ):
|
|
r = Rect( (1,2), (3,4) )
|
|
self.assertEqual( 1, r.left )
|
|
self.assertEqual( 2, r.top )
|
|
self.assertEqual( 3, r.width )
|
|
self.assertEqual( 4, r.height )
|
|
|
|
def testCalculatedAttributes( self ):
|
|
r = Rect( 1, 2, 3, 4 )
|
|
|
|
self.assertEqual( r.left+r.width, r.right )
|
|
self.assertEqual( r.top+r.height, r.bottom )
|
|
self.assertEqual( (r.width,r.height), r.size )
|
|
self.assertEqual( (r.left,r.top), r.topleft )
|
|
self.assertEqual( (r.right,r.top), r.topright )
|
|
self.assertEqual( (r.left,r.bottom), r.bottomleft )
|
|
self.assertEqual( (r.right,r.bottom), r.bottomright )
|
|
|
|
midx = r.left + r.width // 2
|
|
midy = r.top + r.height // 2
|
|
|
|
self.assertEqual( midx, r.centerx )
|
|
self.assertEqual( midy, r.centery )
|
|
self.assertEqual( (r.centerx,r.centery), r.center )
|
|
self.assertEqual( (r.centerx,r.top), r.midtop )
|
|
self.assertEqual( (r.centerx,r.bottom), r.midbottom )
|
|
self.assertEqual( (r.left,r.centery), r.midleft )
|
|
self.assertEqual( (r.right,r.centery), r.midright )
|
|
|
|
def test_normalize( self ):
|
|
r = Rect( 1, 2, -3, -6 )
|
|
r2 = Rect(r)
|
|
r2.normalize()
|
|
self.failUnless( r2.width >= 0 )
|
|
self.failUnless( r2.height >= 0 )
|
|
self.assertEqual( (abs(r.width),abs(r.height)), r2.size )
|
|
self.assertEqual( (-2,-4), r2.topleft )
|
|
|
|
def test_left( self ):
|
|
"""Changing the left attribute moves the rect and does not change
|
|
the rect's width
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_left = 10
|
|
|
|
r.left = new_left
|
|
self.assertEqual( new_left, r.left )
|
|
self.assertEqual( Rect(new_left,2,3,4), r )
|
|
|
|
def test_right( self ):
|
|
"""Changing the right attribute moves the rect and does not change
|
|
the rect's width
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_right = r.right + 20
|
|
expected_left = r.left + 20
|
|
old_width = r.width
|
|
|
|
r.right = new_right
|
|
self.assertEqual( new_right, r.right )
|
|
self.assertEqual( expected_left, r.left )
|
|
self.assertEqual( old_width, r.width )
|
|
|
|
def test_top( self ):
|
|
"""Changing the top attribute moves the rect and does not change
|
|
the rect's width
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_top = 10
|
|
|
|
r.top = new_top
|
|
self.assertEqual( Rect(1,new_top,3,4), r )
|
|
self.assertEqual( new_top, r.top )
|
|
|
|
def test_bottom( self ):
|
|
"""Changing the bottom attribute moves the rect and does not change
|
|
the rect's height
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_bottom = r.bottom + 20
|
|
expected_top = r.top + 20
|
|
old_height = r.height
|
|
|
|
r.bottom = new_bottom
|
|
self.assertEqual( new_bottom, r.bottom )
|
|
self.assertEqual( expected_top, r.top )
|
|
self.assertEqual( old_height, r.height )
|
|
|
|
def test_centerx( self ):
|
|
"""Changing the centerx attribute moves the rect and does not change
|
|
the rect's width
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_centerx = r.centerx + 20
|
|
expected_left = r.left + 20
|
|
old_width = r.width
|
|
|
|
r.centerx = new_centerx
|
|
self.assertEqual( new_centerx, r.centerx )
|
|
self.assertEqual( expected_left, r.left )
|
|
self.assertEqual( old_width, r.width )
|
|
|
|
def test_centery( self ):
|
|
"""Changing the centerx attribute moves the rect and does not change
|
|
the rect's width
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_centery = r.centery + 20
|
|
expected_top = r.top + 20
|
|
old_height = r.height
|
|
|
|
r.centery = new_centery
|
|
self.assertEqual( new_centery, r.centery )
|
|
self.assertEqual( expected_top, r.top )
|
|
self.assertEqual( old_height, r.height )
|
|
|
|
def test_topleft( self ):
|
|
"""Changing the topleft attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.topleft = new_topleft
|
|
self.assertEqual( new_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_bottomleft( self ):
|
|
"""Changing the bottomleft attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_bottomleft = (r.left+20,r.bottom+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.bottomleft = new_bottomleft
|
|
self.assertEqual( new_bottomleft, r.bottomleft )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_topright( self ):
|
|
"""Changing the bottomleft attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_topright = (r.right+20,r.top+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.topright = new_topright
|
|
self.assertEqual( new_topright, r.topright )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_bottomright( self ):
|
|
"""Changing the bottomright attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_bottomright = (r.right+20,r.bottom+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.bottomright = new_bottomright
|
|
self.assertEqual( new_bottomright, r.bottomright )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_center( self ):
|
|
"""Changing the center attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_center = (r.centerx+20,r.centery+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.center = new_center
|
|
self.assertEqual( new_center, r.center )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_midleft( self ):
|
|
"""Changing the midleft attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_midleft = (r.left+20,r.centery+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.midleft = new_midleft
|
|
self.assertEqual( new_midleft, r.midleft )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_midright( self ):
|
|
"""Changing the midright attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_midright= (r.right+20,r.centery+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.midright = new_midright
|
|
self.assertEqual( new_midright, r.midright )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_midtop( self ):
|
|
"""Changing the midtop attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_midtop= (r.centerx+20,r.top+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.midtop = new_midtop
|
|
self.assertEqual( new_midtop, r.midtop )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_midbottom( self ):
|
|
"""Changing the midbottom attribute moves the rect and does not change
|
|
the rect's size
|
|
"""
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_midbottom = (r.centerx+20,r.bottom+30)
|
|
expected_topleft = (r.left+20,r.top+30)
|
|
old_size = r.size
|
|
|
|
r.midbottom = new_midbottom
|
|
self.assertEqual( new_midbottom, r.midbottom )
|
|
self.assertEqual( expected_topleft, r.topleft )
|
|
self.assertEqual( old_size, r.size )
|
|
|
|
def test_width( self ):
|
|
"Changing the width resizes the rect from the top-left corner"
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_width = 10
|
|
old_topleft = r.topleft
|
|
old_height = r.height
|
|
|
|
r.width = new_width
|
|
self.assertEqual( new_width, r.width )
|
|
self.assertEqual( old_height, r.height )
|
|
self.assertEqual( old_topleft, r.topleft )
|
|
|
|
def test_height( self ):
|
|
"Changing the height resizes the rect from the top-left corner"
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_height = 10
|
|
old_topleft = r.topleft
|
|
old_width = r.width
|
|
|
|
r.height = new_height
|
|
self.assertEqual( new_height, r.height )
|
|
self.assertEqual( old_width, r.width )
|
|
self.assertEqual( old_topleft, r.topleft )
|
|
|
|
def test_size( self ):
|
|
"Changing the size resizes the rect from the top-left corner"
|
|
r = Rect( 1, 2, 3, 4 )
|
|
new_size = (10,20)
|
|
old_topleft = r.topleft
|
|
|
|
r.size = new_size
|
|
self.assertEqual( new_size, r.size )
|
|
self.assertEqual( old_topleft, r.topleft )
|
|
|
|
def test_contains( self ):
|
|
r = Rect( 1, 2, 3, 4 )
|
|
|
|
self.failUnless( r.contains( Rect( 2, 3, 1, 1 ) ),
|
|
"r does not contain Rect(2,3,1,1)" )
|
|
self.failUnless( r.contains( Rect(r) ),
|
|
"r does not contain the same rect as itself" )
|
|
self.failUnless( r.contains( Rect(2,3,0,0) ),
|
|
"r does not contain an empty rect within its bounds" )
|
|
self.failIf( r.contains( Rect(0,0,1,2) ),
|
|
"r contains Rect(0,0,1,2)" )
|
|
self.failIf( r.contains( Rect(4,6,1,1) ),
|
|
"r contains Rect(4,6,1,1)" )
|
|
self.failIf( r.contains( Rect(4,6,0,0) ),
|
|
"r contains Rect(4,6,0,0)" )
|
|
|
|
def test_collidepoint( self ):
|
|
r = Rect( 1, 2, 3, 4 )
|
|
|
|
self.failUnless( r.collidepoint( r.left, r.top ),
|
|
"r does not collide with point (left,top)" )
|
|
self.failIf( r.collidepoint( r.left-1, r.top ),
|
|
"r collides with point (left-1,top)" )
|
|
self.failIf( r.collidepoint( r.left, r.top-1 ),
|
|
"r collides with point (left,top-1)" )
|
|
self.failIf( r.collidepoint( r.left-1,r.top-1 ),
|
|
"r collides with point (left-1,top-1)" )
|
|
|
|
self.failUnless( r.collidepoint( r.right-1, r.bottom-1 ),
|
|
"r does not collide with point (right-1,bottom-1)")
|
|
self.failIf( r.collidepoint( r.right, r.bottom ),
|
|
"r collides with point (right,bottom)" )
|
|
self.failIf( r.collidepoint( r.right-1, r.bottom ),
|
|
"r collides with point (right-1,bottom)" )
|
|
self.failIf( r.collidepoint( r.right, r.bottom-1 ),
|
|
"r collides with point (right,bottom-1)" )
|
|
|
|
def test_inflate__larger( self ):
|
|
"The inflate method inflates around the center of the rectangle"
|
|
r = Rect( 2, 4, 6, 8 )
|
|
r2 = r.inflate( 4, 6 )
|
|
|
|
self.assertEqual( r.center, r2.center )
|
|
self.assertEqual( r.left-2, r2.left )
|
|
self.assertEqual( r.top-3, r2.top )
|
|
self.assertEqual( r.right+2, r2.right )
|
|
self.assertEqual( r.bottom+3, r2.bottom )
|
|
self.assertEqual( r.width+4, r2.width )
|
|
self.assertEqual( r.height+6, r2.height )
|
|
|
|
def test_inflate__smaller( self ):
|
|
"The inflate method inflates around the center of the rectangle"
|
|
r = Rect( 2, 4, 6, 8 )
|
|
r2 = r.inflate( -4, -6 )
|
|
|
|
self.assertEqual( r.center, r2.center )
|
|
self.assertEqual( r.left+2, r2.left )
|
|
self.assertEqual( r.top+3, r2.top )
|
|
self.assertEqual( r.right-2, r2.right )
|
|
self.assertEqual( r.bottom-3, r2.bottom )
|
|
self.assertEqual( r.width-4, r2.width )
|
|
self.assertEqual( r.height-6, r2.height )
|
|
|
|
def test_inflate_ip__larger( self ):
|
|
"The inflate_ip method inflates around the center of the rectangle"
|
|
r = Rect( 2, 4, 6, 8 )
|
|
r2 = Rect( r )
|
|
r2.inflate_ip( -4, -6 )
|
|
|
|
self.assertEqual( r.center, r2.center )
|
|
self.assertEqual( r.left+2, r2.left )
|
|
self.assertEqual( r.top+3, r2.top )
|
|
self.assertEqual( r.right-2, r2.right )
|
|
self.assertEqual( r.bottom-3, r2.bottom )
|
|
self.assertEqual( r.width-4, r2.width )
|
|
self.assertEqual( r.height-6, r2.height )
|
|
|
|
def test_inflate_ip__smaller( self ):
|
|
"The inflate method inflates around the center of the rectangle"
|
|
r = Rect( 2, 4, 6, 8 )
|
|
r2 = Rect( r )
|
|
r2.inflate_ip( -4, -6 )
|
|
|
|
self.assertEqual( r.center, r2.center )
|
|
self.assertEqual( r.left+2, r2.left )
|
|
self.assertEqual( r.top+3, r2.top )
|
|
self.assertEqual( r.right-2, r2.right )
|
|
self.assertEqual( r.bottom-3, r2.bottom )
|
|
self.assertEqual( r.width-4, r2.width )
|
|
self.assertEqual( r.height-6, r2.height )
|
|
|
|
def test_clamp( self ):
|
|
r = Rect(10, 10, 10, 10)
|
|
c = Rect(19, 12, 5, 5).clamp(r)
|
|
self.assertEqual(c.right, r.right)
|
|
self.assertEqual(c.top, 12)
|
|
c = Rect(1, 2, 3, 4).clamp(r)
|
|
self.assertEqual(c.topleft, r.topleft)
|
|
c = Rect(5, 500, 22, 33).clamp(r)
|
|
self.assertEqual(c.center, r.center)
|
|
|
|
def test_clamp_ip( self ):
|
|
r = Rect(10, 10, 10, 10)
|
|
c = Rect(19, 12, 5, 5)
|
|
c.clamp_ip(r)
|
|
self.assertEqual(c.right, r.right)
|
|
self.assertEqual(c.top, 12)
|
|
c = Rect(1, 2, 3, 4)
|
|
c.clamp_ip(r)
|
|
self.assertEqual(c.topleft, r.topleft)
|
|
c = Rect(5, 500, 22, 33)
|
|
c.clamp_ip(r)
|
|
self.assertEqual(c.center, r.center)
|
|
|
|
def test_clip( self ):
|
|
r1 = Rect( 1, 2, 3, 4 )
|
|
self.assertEqual( Rect( 1, 2, 2, 2 ), r1.clip( Rect(0,0,3,4) ) )
|
|
self.assertEqual( Rect( 2, 2, 2, 4 ), r1.clip( Rect(2,2,10,20) ) )
|
|
self.assertEqual( Rect(2,3,1,2), r1.clip( Rect(2,3,1,2) ) )
|
|
self.assertEqual( (0,0), r1.clip(20,30,5,6).size )
|
|
self.assertEqual( r1, r1.clip( Rect(r1) ),
|
|
"r1 does not clip an identical rect to itself" )
|
|
|
|
def test_move( self ):
|
|
r = Rect( 1, 2, 3, 4 )
|
|
move_x = 10
|
|
move_y = 20
|
|
r2 = r.move( move_x, move_y )
|
|
expected_r2 = Rect(r.left+move_x,r.top+move_y,r.width,r.height)
|
|
self.assertEqual( expected_r2, r2 )
|
|
|
|
def test_move_ip( self ):
|
|
r = Rect( 1, 2, 3, 4 )
|
|
r2 = Rect( r )
|
|
move_x = 10
|
|
move_y = 20
|
|
r2.move_ip( move_x, move_y )
|
|
expected_r2 = Rect(r.left+move_x,r.top+move_y,r.width,r.height)
|
|
self.assertEqual( expected_r2, r2 )
|
|
|
|
def test_union( self ):
|
|
r1 = Rect( 1, 1, 1, 2 )
|
|
r2 = Rect( -2, -2, 1, 2 )
|
|
self.assertEqual( Rect( -2, -2, 4, 5 ), r1.union(r2) )
|
|
|
|
def test_union__with_identical_Rect( self ):
|
|
r1 = Rect( 1, 2, 3, 4 )
|
|
self.assertEqual( r1, r1.union( Rect(r1) ) )
|
|
|
|
def test_union_ip( self ):
|
|
r1 = Rect( 1, 1, 1, 2 )
|
|
r2 = Rect( -2, -2, 1, 2 )
|
|
r1.union_ip(r2)
|
|
self.assertEqual( Rect( -2, -2, 4, 5 ), r1 )
|
|
|
|
def test_unionall( self ):
|
|
r1 = Rect( 0, 0, 1, 1 )
|
|
r2 = Rect( -2, -2, 1, 1 )
|
|
r3 = Rect( 2, 2, 1, 1 )
|
|
|
|
r4 = r1.unionall( [r2,r3] )
|
|
self.assertEqual( Rect(-2, -2, 5, 5), r4 )
|
|
|
|
def test_unionall_ip( self ):
|
|
r1 = Rect( 0, 0, 1, 1 )
|
|
r2 = Rect( -2, -2, 1, 1 )
|
|
r3 = Rect( 2, 2, 1, 1 )
|
|
|
|
r1.unionall_ip( [r2,r3] )
|
|
self.assertEqual( Rect(-2, -2, 5, 5), r1 )
|
|
|
|
# Bug for an empty list. Would return a Rect instead of None.
|
|
self.assertTrue(r1.unionall_ip( [] ) is None)
|
|
|
|
def test_colliderect( self ):
|
|
r1 = Rect(1,2,3,4)
|
|
self.failUnless( r1.colliderect( Rect(0,0,2,3) ),
|
|
"r1 does not collide with Rect(0,0,2,3)" )
|
|
self.failIf( r1.colliderect( Rect(0,0,1,2) ),
|
|
"r1 collides with Rect(0,0,1,2)" )
|
|
self.failIf( r1.colliderect( Rect(r1.right,r1.bottom,2,2) ),
|
|
"r1 collides with Rect(r1.right,r1.bottom,2,2)" )
|
|
self.failUnless( r1.colliderect( Rect(r1.left+1,r1.top+1,
|
|
r1.width-2,r1.height-2) ),
|
|
"r1 does not collide with Rect(r1.left+1,r1.top+1,"+
|
|
"r1.width-2,r1.height-2)" )
|
|
self.failUnless( r1.colliderect( Rect(r1.left-1,r1.top-1,
|
|
r1.width+2,r1.height+2) ),
|
|
"r1 does not collide with Rect(r1.left-1,r1.top-1,"+
|
|
"r1.width+2,r1.height+2)" )
|
|
self.failUnless( r1.colliderect( Rect(r1) ),
|
|
"r1 does not collide with an identical rect" )
|
|
self.failIf( r1.colliderect( Rect(r1.right,r1.bottom,0,0) ),
|
|
"r1 collides with Rect(r1.right,r1.bottom,0,0)" )
|
|
self.failIf( r1.colliderect( Rect(r1.right,r1.bottom,1,1) ),
|
|
"r1 collides with Rect(r1.right,r1.bottom,1,1)" )
|
|
|
|
def testEquals( self ):
|
|
""" check to see how the rect uses __eq__
|
|
"""
|
|
r1 = Rect(1,2,3,4)
|
|
r2 = Rect(10,20,30,40)
|
|
r3 = (10,20,30,40)
|
|
r4 = Rect(10,20,30,40)
|
|
|
|
class foo (Rect):
|
|
def __eq__(self,other):
|
|
return id(self) == id(other)
|
|
def __ne__(self,other):
|
|
return id(self) != id(other)
|
|
|
|
class foo2 (Rect):
|
|
pass
|
|
|
|
r5 = foo(10,20,30,40)
|
|
r6 = foo2(10,20,30,40)
|
|
|
|
self.assertNotEqual(r5, r2)
|
|
|
|
# because we define equality differently for this subclass.
|
|
self.assertEqual(r6, r2)
|
|
|
|
|
|
rect_list = [r1,r2,r3,r4,r6]
|
|
|
|
# see if we can remove 4 of these.
|
|
rect_list.remove(r2)
|
|
rect_list.remove(r2)
|
|
rect_list.remove(r2)
|
|
rect_list.remove(r2)
|
|
self.assertRaises(ValueError, rect_list.remove, r2)
|
|
|
|
def test_collidedict(self):
|
|
|
|
# __doc__ (as of 2008-08-02) for pygame.rect.Rect.collidedict:
|
|
|
|
# Rect.collidedict(dict): return (key, value)
|
|
# test if one rectangle in a dictionary intersects
|
|
#
|
|
# Returns the key and value of the first dictionary value that
|
|
# collides with the Rect. If no collisions are found, None is
|
|
# returned.
|
|
#
|
|
# Rect objects are not hashable and cannot be used as keys in a
|
|
# dictionary, only as values.
|
|
|
|
r = Rect(1, 1, 10, 10)
|
|
r1 = Rect(1, 1, 10, 10)
|
|
r2 = Rect(50, 50, 10, 10)
|
|
r3 = Rect(70, 70, 10, 10)
|
|
r4 = Rect(61, 61, 10, 10)
|
|
|
|
d = {1: r1, 2: r2, 3: r3}
|
|
|
|
rects_values = 1
|
|
val = r.collidedict(d, rects_values)
|
|
self.assertTrue(val)
|
|
self.assertEqual(len(val), 2)
|
|
self.assertEqual(val[0], 1)
|
|
self.assertEqual(val[1], r1)
|
|
|
|
none_d = {2: r2, 3: r3}
|
|
none_val = r.collidedict(none_d, rects_values)
|
|
self.assertFalse(none_val)
|
|
|
|
barely_d = {1: r1, 2: r2, 3: r3}
|
|
k3, v3 = r4.collidedict(barely_d, rects_values)
|
|
self.assertEqual(k3, 3)
|
|
self.assertEqual(v3, r3)
|
|
|
|
|
|
def test_collidedictall(self):
|
|
|
|
# __doc__ (as of 2008-08-02) for pygame.rect.Rect.collidedictall:
|
|
|
|
# Rect.collidedictall(dict): return [(key, value), ...]
|
|
# test if all rectangles in a dictionary intersect
|
|
#
|
|
# Returns a list of all the key and value pairs that intersect with
|
|
# the Rect. If no collisions are found an empty dictionary is
|
|
# returned.
|
|
#
|
|
# Rect objects are not hashable and cannot be used as keys in a
|
|
# dictionary, only as values.
|
|
|
|
r = Rect(1, 1, 10, 10)
|
|
|
|
r2 = Rect(1, 1, 10, 10)
|
|
r3 = Rect(5, 5, 10, 10)
|
|
r4 = Rect(10, 10, 10, 10)
|
|
r5 = Rect(50, 50, 10, 10)
|
|
|
|
rects_values = 1
|
|
d = {2: r2}
|
|
l = r.collidedictall(d, rects_values)
|
|
self.assertEqual(l, [(2, r2)])
|
|
|
|
d2 = {2: r2, 3: r3, 4: r4, 5: r5}
|
|
l2 = r.collidedictall(d2, rects_values)
|
|
self.assertEqual(l2, [(2, r2), (3, r3), (4, r4)])
|
|
|
|
def test_collidelist(self):
|
|
|
|
# __doc__ (as of 2008-08-02) for pygame.rect.Rect.collidelist:
|
|
|
|
# Rect.collidelist(list): return index
|
|
# test if one rectangle in a list intersects
|
|
#
|
|
# Test whether the rectangle collides with any in a sequence of
|
|
# rectangles. The index of the first collision found is returned. If
|
|
# no collisions are found an index of -1 is returned.
|
|
|
|
r = Rect(1, 1, 10, 10)
|
|
l = [Rect(50, 50, 1, 1), Rect(5, 5, 10, 10), Rect(15, 15, 1, 1)]
|
|
|
|
self.assertEqual(r.collidelist(l), 1)
|
|
|
|
f = [Rect(50, 50, 1, 1), (100, 100, 4, 4)]
|
|
self.assertEqual(r.collidelist(f), -1)
|
|
|
|
|
|
def test_collidelistall(self):
|
|
|
|
# __doc__ (as of 2008-08-02) for pygame.rect.Rect.collidelistall:
|
|
|
|
# Rect.collidelistall(list): return indices
|
|
# test if all rectangles in a list intersect
|
|
#
|
|
# Returns a list of all the indices that contain rectangles that
|
|
# collide with the Rect. If no intersecting rectangles are found, an
|
|
# empty list is returned.
|
|
|
|
r = Rect(1, 1, 10, 10)
|
|
|
|
l = [
|
|
Rect(1, 1, 10, 10),
|
|
Rect(5, 5, 10, 10),
|
|
Rect(15, 15, 1, 1),
|
|
Rect(2, 2, 1, 1),
|
|
]
|
|
self.assertEqual(r.collidelistall(l), [0, 1, 3])
|
|
|
|
f = [Rect(50, 50, 1, 1), Rect(20, 20, 5, 5)]
|
|
self.assertFalse(r.collidelistall(f))
|
|
|
|
|
|
def test_fit(self):
|
|
|
|
# __doc__ (as of 2008-08-02) for pygame.rect.Rect.fit:
|
|
|
|
# Rect.fit(Rect): return Rect
|
|
# resize and move a rectangle with aspect ratio
|
|
#
|
|
# Returns a new rectangle that is moved and resized to fit another.
|
|
# The aspect ratio of the original Rect is preserved, so the new
|
|
# rectangle may be smaller than the target in either width or height.
|
|
|
|
r = Rect(10, 10, 30, 30)
|
|
|
|
r2 = Rect(30, 30, 15, 10)
|
|
|
|
f = r.fit(r2)
|
|
self.assertTrue(r2.contains(f))
|
|
|
|
f2 = r2.fit(r)
|
|
self.assertTrue(r.contains(f2))
|
|
|
|
|
|
|
|
def test_copy(self):
|
|
r = Rect(1, 2, 10, 20)
|
|
c = r.copy()
|
|
self.failUnlessEqual(c, r)
|
|
|
|
|
|
def test_subscript(self):
|
|
r = Rect(1, 2, 3, 4)
|
|
self.assertEqual(r[0], 1)
|
|
self.assertEqual(r[1], 2)
|
|
self.assertEqual(r[2], 3)
|
|
self.assertEqual(r[3], 4)
|
|
self.assertEqual(r[-1], 4)
|
|
self.assertEqual(r[-2], 3)
|
|
self.assertEqual(r[-4], 1)
|
|
self.assertRaises(IndexError, r.__getitem__, 5)
|
|
self.assertRaises(IndexError, r.__getitem__, -5)
|
|
self.assertEqual(r[0:2], [1, 2])
|
|
self.assertEqual(r[0:4], [1, 2, 3, 4])
|
|
self.assertEqual(r[0:-1], [1, 2, 3])
|
|
self.assertEqual(r[:], [1, 2, 3, 4])
|
|
self.assertEqual(r[...], [1, 2, 3, 4])
|
|
self.assertEqual(r[0:4:2], [1, 3])
|
|
self.assertEqual(r[0:4:3], [1, 4])
|
|
self.assertEqual(r[3::-1], [4, 3, 2, 1])
|
|
self.assertRaises(TypeError, r.__getitem__, None)
|
|
|
|
def test_ass_subscript(self):
|
|
r = Rect(0, 0, 0, 0)
|
|
r[...] = 1, 2, 3, 4
|
|
self.assertEqual(r, [1, 2, 3, 4])
|
|
self.assertRaises(TypeError, r.__setitem__, None, 0)
|
|
self.assertEqual(r, [1, 2, 3, 4])
|
|
self.assertRaises(TypeError, r.__setitem__, 0, '')
|
|
self.assertEqual(r, [1, 2, 3, 4])
|
|
self.assertRaises(IndexError, r.__setitem__, 4, 0)
|
|
self.assertEqual(r, [1, 2, 3, 4])
|
|
self.assertRaises(IndexError, r.__setitem__, -5, 0)
|
|
self.assertEqual(r, [1, 2, 3, 4])
|
|
r[0] = 10
|
|
self.assertEqual(r, [10, 2, 3, 4])
|
|
r[3] = 40
|
|
self.assertEqual(r, [10, 2, 3, 40])
|
|
r[-1] = 400
|
|
self.assertEqual(r, [10, 2, 3, 400])
|
|
r[-4] = 100
|
|
self.assertEqual(r, [100, 2, 3, 400])
|
|
r[1:3] = 0
|
|
self.assertEqual(r, [100, 0, 0, 400])
|
|
r[...] = 0
|
|
self.assertEqual(r, [0, 0, 0, 0])
|
|
r[:] = 9
|
|
self.assertEqual(r, [9, 9, 9, 9])
|
|
r[:] = 11, 12, 13, 14
|
|
self.assertEqual(r, [11, 12, 13, 14])
|
|
r[::-1] = r
|
|
self.assertEqual(r, [14, 13, 12, 11])
|
|
|
|
|
|
class SubclassTest(unittest.TestCase):
|
|
class MyRect(Rect):
|
|
def __init__(self, *args, **kwds):
|
|
super(SubclassTest.MyRect, self).__init__(*args, **kwds)
|
|
self.an_attribute = True
|
|
|
|
def test_copy(self):
|
|
mr1 = self.MyRect(1, 2, 10, 20)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.copy()
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_move(self):
|
|
mr1 = self.MyRect(1, 2, 10, 20)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.move(1, 2)
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_inflate(self):
|
|
mr1 = self.MyRect(1, 2, 10, 20)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.inflate(2, 4)
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_clamp(self):
|
|
mr1 = self.MyRect(19, 12, 5, 5)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.clamp(Rect(10, 10, 10, 10))
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_clip(self):
|
|
mr1 = self.MyRect(1, 2, 3, 4)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.clip(Rect(0, 0, 3, 4))
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_union(self):
|
|
mr1 = self.MyRect(1, 1, 1, 2)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.union(Rect(-2, -2, 1, 2))
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_unionall(self):
|
|
mr1 = self.MyRect(0, 0, 1, 1)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.unionall([Rect(-2, -2, 1, 1), Rect(2, 2, 1, 1)])
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
def test_fit(self):
|
|
mr1 = self.MyRect(10, 10, 30, 30)
|
|
self.assertTrue(mr1.an_attribute)
|
|
mr2 = mr1.fit(Rect(30, 30, 15, 10))
|
|
self.assertTrue(isinstance(mr2, self.MyRect))
|
|
self.assertRaises(AttributeError, getattr, mr2, "an_attribute")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|