예제 #1
0
class BulletsTestCase(unittest.TestCase):
    """ Tests the wrapper of sprite group of bullets """
    def setUp(self):
        self.bullets = Bullets()

    def test_initialization(self):
        self.assertIsNotNone(self.bullets.get_bullets())
        self.assertEqual(len(self.bullets.get_bullets()), 0)

    def test_maximum_threshold(self):
        for i in range(11):
            x = random.randint(0, 200)
            y = random.randint(0, 200)

            self.bullets.add(Bullet(x, y))

        self.assertEqual(len(self.bullets.get_bullets()), 10)

    def test_out_of_screen_bullet(self):
        bullets = Bullets()
        bullet_out_of_screen = Bullet(400, -1)

        bullets.add(bullet_out_of_screen)

        bullets.update()

        self.assertEqual(len(bullets.get_bullets()), 0)
예제 #2
0
    def test_out_of_screen_bullet(self):
        bullets = Bullets()
        bullet_out_of_screen = Bullet(400, -1)

        bullets.add(bullet_out_of_screen)

        bullets.update()

        self.assertEqual(len(bullets.get_bullets()), 0)