def test_equal_10(self):
characters = '54321'
expected_watch_from_side = dedent('''\
1
222
33333
4444444
555555555''')
expected_watch_from_above = dedent('''\
555555555
544444445
543333345
543222345
543212345
543222345
543333345
544444445
555555555''')
self.assertEqual(
count_visible_characters_of_the_pyramid(characters), 81
)
self.assertEqual(count_all_characters_of_the_pyramid(characters), 165)
self.assertEqual(watch_pyramid_from_the_side(characters),
expected_watch_from_side)
self.assertEqual(watch_pyramid_from_above(characters),
expected_watch_from_above)
def test_equal_7(self):
characters = '*#'
expected_watch_from_side = dedent('''\
#
***''')
expected_watch_from_above = dedent('''\
***
*#*
***''')
self.assertEqual(
count_visible_characters_of_the_pyramid(characters), 9
)
self.assertEqual(count_all_characters_of_the_pyramid(characters), 10)
self.assertEqual(watch_pyramid_from_the_side(characters),
expected_watch_from_side)
self.assertEqual(watch_pyramid_from_above(characters),
expected_watch_from_above)
def test_equal_9(self):
characters = 'aaa'
expected_watch_from_side = dedent('''\
a
aaa
aaaaa''')
expected_watch_from_above = dedent('''\
aaaaa
aaaaa
aaaaa
aaaaa
aaaaa''')
self.assertEqual(
count_visible_characters_of_the_pyramid(characters), 25
)
self.assertEqual(count_all_characters_of_the_pyramid(characters), 35)
self.assertEqual(watch_pyramid_from_the_side(characters),
expected_watch_from_side)
self.assertEqual(watch_pyramid_from_above(characters),
expected_watch_from_above)
def test_equal_5(self):
self.assertEqual(count_visible_characters_of_the_pyramid(''), -1)
