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)