def test_get_top_level_returns_the_last_element_form_the_level_list(self): pyr = Pyramid(10, 3) pyr.levels = [12, 13, 14] self.assertEqual(pyr.get_top_level(), 14)
def test_get_level_returns_the_correct_list_element(self): pyr = Pyramid(10, 3) pyr.levels = [12, 13, 14] self.assertEqual(pyr.get_level(1), 13) self.assertEqual(pyr.get_level(0), 12)