def setUp(self):
self.ag = sprite.AbstractGroup()
self.ag2 = sprite.AbstractGroup()
self.s1 = sprite.Sprite(self.ag)
self.s2 = sprite.Sprite(self.ag)
self.s3 = sprite.Sprite(self.ag2)
self.s4 = sprite.Sprite(self.ag2)
self.s1.image = pygame.Surface((10, 10))
self.s1.image.fill(pygame.Color("red"))
self.s1.rect = self.s1.image.get_rect()
self.s2.image = pygame.Surface((10, 10))
self.s2.image.fill(pygame.Color("green"))
self.s2.rect = self.s2.image.get_rect()
self.s2.rect.left = 10
self.s3.image = pygame.Surface((10, 10))
self.s3.image.fill(pygame.Color("blue"))
self.s3.rect = self.s3.image.get_rect()
self.s3.rect.top = 10
self.s4.image = pygame.Surface((10, 10))
self.s4.image.fill(pygame.Color("white"))
self.s4.rect = self.s4.image.get_rect()
self.s4.rect.left = 10
self.s4.rect.top = 10
self.bg = pygame.Surface((20, 20))
self.scr = pygame.Surface((20, 20))
self.scr.fill(pygame.Color("grey"))
def setUp(self):
self.ag = sprite.AbstractGroup()
self.ag2 = sprite.AbstractGroup()
self.s1 = sprite.Sprite(self.ag)
self.s2 = sprite.Sprite(self.ag)
self.s3 = sprite.Sprite(self.ag2)
self.s4 = sprite.Sprite(self.ag2)
self.s1.image = pygame.Surface((10, 10))
self.s1.image.fill(pygame.Color('red'))
self.s1.rect = self.s1.image.get_rect()
self.s2.image = pygame.Surface((10, 10))
self.s2.image.fill(pygame.Color('green'))
self.s2.rect = self.s2.image.get_rect()
self.s2.rect.move_left = 10
self.s3.image = pygame.Surface((10, 10))
self.s3.image.fill(pygame.Color('blue'))
self.s3.rect = self.s3.image.get_rect()
self.s3.rect.top = 10
self.s4.image = pygame.Surface((10, 10))
self.s4.image.fill(pygame.Color('white'))
self.s4.rect = self.s4.image.get_rect()
self.s4.rect.move_left = 10
self.s4.rect.top = 10
self.bg = pygame.Surface((20, 20))
self.scr = pygame.Surface((20, 20))
self.scr.fill(pygame.Color('grey'))
def setUp(self):
self.ag = sprite.AbstractGroup()
self.ag2 = sprite.AbstractGroup()
self.s1 = sprite.Sprite(self.ag)
self.s2 = sprite.Sprite(self.ag2)
self.s3 = sprite.Sprite(self.ag2)
self.s1.image = pygame.Surface((50, 10), pygame.SRCALPHA, 32)
self.s2.image = pygame.Surface((10, 10), pygame.SRCALPHA, 32)
self.s3.image = pygame.Surface((10, 10), pygame.SRCALPHA, 32)
self.s1.rect = self.s1.image.get_rect()
self.s2.rect = self.s2.image.get_rect()
self.s3.rect = self.s3.image.get_rect()
self.s2.rect.move_ip(40, 0)
self.s3.rect.move_ip(100, 100)
def test_add(self):
ag3 = sprite.AbstractGroup()
sprites = (self.s1, self.s2, self.s3, self.s4)
for s in sprites:
self.assertNotIn(s, ag3)
ag3.add(self.s1, [self.s2], self.ag2)
for s in sprites:
self.assertIn(s, ag3)
def test_has(self):
" See if AbstractGroup.has() works as expected. "
ag = sprite.AbstractGroup()
ag2 = sprite.AbstractGroup()
s1 = sprite.Sprite(ag)
s2 = sprite.Sprite(ag)
s3 = sprite.Sprite(ag2)
s4 = sprite.Sprite(ag2)
self.assertEqual(True, s1 in ag)
self.assertEqual(True, ag.has(s1))
self.assertEqual(True, ag.has([s1, s2]))
# see if one of them not being in there.
self.assertNotEqual(True, ag.has([s1, s2, s3]))
# see if a second AbstractGroup works.
self.assertEqual(True, ag2.has(s3))
def test_add(self):
ag3 = sprite.AbstractGroup()
self.assertFalse(self.s1 in ag3)
self.assertFalse(self.s2 in ag3)
self.assertFalse(self.s3 in ag3)
self.assertFalse(self.s4 in ag3)
ag3.add(self.s1, [self.s2], self.ag2)
self.assert_(self.s1 in ag3)
self.assert_(self.s2 in ag3)
self.assert_(self.s3 in ag3)
self.assert_(self.s4 in ag3)
