import unittest,test_reader,data.datareader
from src.familytree import FamilyTree
from src.treecreator import create_tree
# Create family tree
tree_version = 'OriginalTree'
family_tree,error = create_tree('OriginalTree')
class OriginalTreeTests(unittest.TestCase):
def test_get_grandparents(self):
test_data = test_reader.get_family_tree_tests(tree_version,'get_grandparents')
for test in test_data:
expected_result = test_data[test]
test_result = family_tree.get_grandparent(test)
self.assertEqual(test_result,expected_result,msg='{} failure'.format(input))
def test_get_people_without_siblings(self):
expected_result = test_reader.get_family_tree_tests(tree_version,'get_people_without_siblings')
test_result = family_tree.get_people_without_siblings()
self.assertEqual(test_result,expected_result,msg='{} failure'.format('No sibling test'))
def test_get_people_without_children(self):
expected_result = test_reader.get_family_tree_tests(tree_version,'get_people_without_children')
test_result = family_tree.get_people_without_children()
self.assertEqual(test_result,expected_result,msg='{} failure'.format('No children test'))
def test_get_person_with_largest_number_of_grandchildren(self):
expected_result = test_reader.get_family_tree_tests(tree_version,'get_person_with_most_grandchildren')
test_result = family_tree.get_person_with_largest_number_of_grandchildren()
self.assertEqual(test_result,expected_result,msg='{} failure'.format('Most grandchildren test'))
if __name__ == '__main__':
unittest.main()
import unittest, test_reader, data.datareader
from src.familytree import FamilyTree
from src.treecreator import create_tree
# Create family tree
tree_version = "TrialTree"
family_tree, error = create_tree(tree_version)
class TrialTreeTests(unittest.TestCase):
def test_get_people_without_siblings(self):
expected_result = test_reader.get_family_tree_tests(tree_version, "get_people_without_siblings")
test_result = family_tree.get_people_without_siblings()
self.assertEqual(test_result, expected_result, msg="{} failure".format("No sibling test"))
def test_get_person_with_largest_number_of_grandchildren(self):
expected_result = test_reader.get_family_tree_tests(tree_version, "get_person_with_most_grandchildren")
test_result = family_tree.get_person_with_largest_number_of_grandchildren()
self.assertEqual(test_result, expected_result, msg="{} failure".format("Most grandchildren test"))
if __name__ == "__main__":
unittest.main()
