def test_empty_ways(self):
input_string = ""
people = string_to_people(input_string)
tree = Tree()
for person in people:
tree.add_person(person)
ways = tree.do_it()
self.assertEqual(0, ways)
def test_counting_ways(self):
input_string = "YNN YNY YNY NYY NYY NYN"
people = string_to_people(input_string)
tree = Tree()
for person in people:
tree.add_person(person)
ways = tree.do_it()
self.assertEqual(2, ways)
from Tree import Tree, string_to_people
input_string = "YNNNYNNYNYNYNNYNNNNYNNNYYYNNNNNYNNNYNNNNNNNNNNNNYN NYNNYYYNNNNNNNNNNNNYNNNNNNNYNNYNNNNYNNNYNNNNNNNNNY NNNNNNNNNNNYNNNNYNNNNNNNNNNNNNNNNNNYYNYNNNNNNNNNNN YNNNNNNNNNYNNNYNNNNNYNNNNNYYNNNNNNNNNNYNNNNYNNNYNN NNNYNNYNNNNNNNNYNNNNNYNYNNYNYNNNNNNNNNNNYYNNNNYNNY YNNNNNNNNNNNYNNNNNYNYNYNNNYNNNNNNNYNNYNNNNNNNNNYNN YNNYNNYYNYYNYNNNNYNNNNYNNNNNNNNYYNYNYYNNNNNNNNNNNY YYYNYNYNNYNNYNNNNNYNNNNNNNNNNNNNYNNNNNYNYNNNNNNNNN YNNNNYNYNNNNNNNNNNNNYNNNNNNNYNNNNNNNNNYNNNNNNYNNNN YNYNNYNNNNNNNNNNNYNNNNYYYNNNNNNYNNNNNYNYYNNYNNYNYN YNNNNNNNYNYYNNNNYNYNNNNNNNYNNNNNNNNYYYNYNNNYNYNNYN NNYNNYYNNNNNNNNNNNYNYYYNNYYNNYNNNNNNNYNNNNNNNYNNYY NNYNYNYNYNYNNNNNNNNNNNNNYYNNNYYNYNNNYYNNYNNNNNNYNN NYNNNNYNNNNNNNNNYNNNNNYNNNYNNNNNNYNNNNNYNNNNNNNNNN NNYYNNNNNNNYNNNNNNNYNNNYNNNNNNNNNNNNNNNNNNNNNYNNNY NNNNNNYNNNNNYNNNNNNNYYNNNYNYYNNNNNYNNNNNNYNNNNNNNN NNNNNNNNYNNNNNNNNNNYNNNNNNNNNYYYNNNNNNNNNNNYNNNNNN NNNNYNNNYNNNYNYNYNNNNNNNNNNNYYNNNNNNNNNNYNNNNNNNYN NNNNNNYNNNNYYNNNNNNNNNNNYNNYNNNNNNNNYNNNYNNNNNNNNN NYYYNNNYNNNNNNYNNYYNNNNNNNYYNNYNNNNNYNNNYNNNYNYNYN YNNYNNYYYNNYYNNYNYNNNNNNNNNNNNNYNNYNNNYNNNNNYYNNNN NYNNNNYNYNNNNNNNNNNNYYYNNNNNNNNNNYNYNNNYNNNYNNNNNN NNNNYNYNNNYYNNNNNNNYNNNYNYNYNNNNNNNNNNYYYNYYNNNNNN NNNNYNYYNNNNNNNNNNYNNNYNYYNYYNYNNYNNNNNNNYNYNYNNNN YNNNYNNNNNNNNNNNNNNNNNYYNNNNYYNNNNNNYYNNYNYYYNNNYN NNNYNNYYYNNYNNNYNNNNNYNNNNNNNNNNNNNNNNNNNNNYNNNNNN NNNNNNYNNNYNNNNNYNNYYNNNNNYNNNYNYNNNNNNNNNNNNNNNYN NNNYNNNNYNNYNNNNNYNNYNNNNNYNNNNNNNYNNNNNYNNNNNNYNY NNNNNYNNNNNNNNNNYNNNNYYNNYYNNYNNYYNYNNNNNNNNNNYYYN YYNYYNYNNNNNNYNNYNNYNNNYNNYNNNNNNYYNNNNNNNNNYYNNYN YNNYNNYNNNYYNNNNYYNNYNNNNNNYNNNYYNNNYYNNYNNNNNNNNN NYNNYYNNNNYNNYYNNYYNYNYNNYNNNNNYNNNYNNNNYNNNNNNNNN NNNNNNNNNNNNYNNNNNNNNNNNNNNYNNNNYNNNNNYYNNNNNNNYNY NYYNNNYNNNNNYYNNNNNNNYNNNNNNYYNYNNNNYNNNNNNNNYNNNN NNYNNNNNNNNNNNYNNNYNNNYNNNNNYNYNNYNNYYNNNNNNNYNNNN NNNNNNNYNNNNYYYYNNNNYNNYYNNYNYYNNNNNYNNNNNNNNNNNNN YNNNNNNNNNNNNNNYNYNNNNNNNNNNNNNNNNNNNNNNNNYYNNNNNN NNYYNYNNYNYYYNYNNNNNNNYYNNNNNNNNNNNNNNNNNNNNNNNNNN NNNNNNNNNYYNYNYYNNYYNNNNNNNNNNNNNNNNYYNYNNYNNNNNNN YNNNYNNNNNNNNNNNNNNNNNNNNNNNNNNYNYNNNNNNNYNNNNNNYY NNNYNNNNYNYNYYYNNNNYNNYNNNNYYYYNNNYYNNNNYNNNNNNNNY NNNNNYNYNNNNYNNYNNNNNYNNNNNNNYNYNNNNNNNNNNNNYNNNNN YNYNNNNNNNNNNNNYYNYNNNYNNNNNYNYNYYYYYYNNNYYNNYNNNY NNNNYNNNNNNNYNNNNYYNNNNNNNNNNNNNYNYYNNYYNNNNNNYNYY NNNNNNNNNNNNNNNNYNYNNNYNNNNNYNNNNYNNNNNNYNNNYNNYNN YNYNNNNNNNYNNNYYNNNNNNNNNNNNYNNNNNNNNNNYYNNNYNNNNN NNNNYNYNNYNYNNNNYNNNNNYNNNNYYYNNNNNNYNYNYNNNNYNNNN NNNNYNNNNNNNNNNNNNYNYNNNNNNNNNYNNNYYYNYNYNNNNNNNYN NNNYNNYNNYYNNNNNNNNYNNYNNYNNYYNNNNNNNNNNNNNYNNNNNN YNNNYNNNNNYNNNNNYNYNNNNYYYNNNNNNNNNNYNNNNYNNNNNYYN"
people = string_to_people(input_string)
tree = Tree()
for person in people:
tree.add_person(person)
for person in tree.people:
print(person)
print(tree.do_it())
