def test_draw_pyramid_height_5(self):
expected_result = (" *\n"
" ***\n"
" *****\n"
" *******\n"
"*********")
result = exercises.draw_pyramid(5)
self.assertEqual(expected_result, result)
def test_draw_pyramid_height_1(self):
expected_result = "*"
result = exercises.draw_pyramid(1)
self.assertEqual(expected_result, result)