class testTerrain(unittest.TestCase): def setUp(self): pygame.init() pygame.display.set_mode([Settings.SCREEN_WIDTH, Settings.SCREEN_HEIGHT]) self.terrain = Terrain() def testAddTerrainBlock(self): self.terrain.addBlock(100, 150) self.assertEqual(len(self.terrain.sprites()), 1) self.assertEqual(self.terrain.sprites()[0].rect.x, 100) self.assertEqual(self.terrain.sprites()[0].rect.y, 150) def testCreateTerrain(self): self.terrain.createEastBorder(5) self.terrain.createNorthBorder(5) self.terrain.createSouthBorder(5) self.terrain.createWestBorder(5) self.assertEqual(len(self.terrain.sprites()), 5 * 4)