def test_staircases_generate_levels(self): self.config['game']['levels'] = 3 game = Game(self.config, []) for level in range(0, self.config['game']['levels']): game.descend_stairs() self.assertEqual(3, game.current_level) game.ascend_stairs() self.assertEqual(2, game.current_level)