class TestMember(TestCase): def setUp(self): self.member = Member(1, "Zim", "Male") self.mother = Member(2, "Mother", "Female") self.father = Member(3, "Dad", "Male") self.mothers_sister_a = Member(4, "MaternalAuntA", "Female") self.mothers_sister_b = Member(5, "MaternalAuntB", "Female") self.mothers_brother_a = Member(6, "MaternalUncleA", "Male") self.mothers_brother_b = Member(7, "MaternalUncleB", "Male") self.fathers_sister_a = Member(8, "PaternalAuntA", "Female") self.fathers_sister_b = Member(9, "PaternalAuntB", "Female") self.fathers_brother_a = Member(10, "PaternalUncleA", "Male") self.fathers_brother_b = Member(11, "PaternalUncleB", "Male") self.spouse = Member(12, "Wife", "Female") self.brother_a = Member(13, "BrotherA", "Male") self.brother_b = Member(14, "BrotherB", "Male") self.sister_a = Member(15, "SisterA", "Female") self.sister_b = Member(16, "SisterB", "Female") self.son_a = Member(17, "SonA", "Male") self.son_b = Member(18, "SonB", "Male") self.daughter_a = Member(19, "DaughterA", "Female") self.daughter_b = Member(20, "DaughterB", "Female") self.paternal_grandmother = Member(21, "PaternalGrandmother", "Female") self.maternal_grandmother = Member(22, "MaternalGrandmother", "Female") # adding our parents self.member.set_mother(self.mother) self.member.set_father(self.father) # adding our siblings self.father.add_child(self.brother_a) self.father.add_child(self.brother_b) self.father.add_child(self.sister_a) self.father.add_child(self.sister_b) self.father.add_child(self.member) self.mother.add_child(self.brother_a) self.mother.add_child(self.brother_b) self.mother.add_child(self.sister_a) self.mother.add_child(self.sister_b) self.mother.add_child(self.member) # add spouse self.member.set_spouse(self.spouse) self.spouse.set_spouse(self.member) # adding our paternal aunt/uncle s self.paternal_grandmother.add_child(self.fathers_sister_a) self.paternal_grandmother.add_child(self.fathers_sister_b) self.paternal_grandmother.add_child(self.fathers_brother_a) self.paternal_grandmother.add_child(self.fathers_brother_b) self.paternal_grandmother.add_child(self.father) self.father.set_mother(self.paternal_grandmother) # adding our maternal aunt/uncle s self.maternal_grandmother.add_child(self.mothers_sister_a) self.maternal_grandmother.add_child(self.mothers_sister_b) self.maternal_grandmother.add_child(self.mothers_brother_a) self.maternal_grandmother.add_child(self.mothers_brother_b) self.maternal_grandmother.add_child(self.mother) self.mother.set_mother(self.maternal_grandmother) # adding our sons and daughters self.member.add_child(self.son_a) self.member.add_child(self.son_b) self.member.add_child(self.daughter_a) self.member.add_child(self.daughter_b) def test_set_methods(self): # test parents self.assertEqual(self.member.mother.name, "Mother") self.assertEqual(self.member.father.name, "Dad") self.assertEqual(self.member in self.member.father.children, True) self.assertEqual(self.member in self.member.mother.children, True) # test siblings self.assertEqual(len(self.member.mother.children), 5) self.assertEqual(self.brother_a in self.member.mother.children, True) self.assertEqual(self.brother_b in self.member.mother.children, True) self.assertEqual(self.sister_a in self.member.mother.children, True) self.assertEqual(self.sister_b in self.member.mother.children, True) self.assertEqual(len(self.member.father.children), 5) self.assertEqual(self.brother_a in self.member.father.children, True) self.assertEqual(self.brother_b in self.member.father.children, True) self.assertEqual(self.sister_a in self.member.father.children, True) self.assertEqual(self.sister_b in self.member.father.children, True) # test spouse self.assertEqual(self.member.spouse.name, "Wife") # test paternal/maternal aunts and uncles self.assertEqual(len(self.member.mother.mother.children), 5) self.assertEqual( self.mothers_brother_a in self.member.mother.mother.children, True) self.assertEqual( self.mothers_brother_b in self.member.mother.mother.children, True) self.assertEqual( self.mothers_sister_a in self.member.mother.mother.children, True) self.assertEqual( self.mothers_sister_b in self.member.mother.mother.children, True) self.assertEqual(self.mother in self.member.mother.mother.children, True) self.assertEqual(len(self.member.father.mother.children), 5) self.assertEqual( self.fathers_brother_a in self.member.father.mother.children, True) self.assertEqual( self.fathers_brother_b in self.member.father.mother.children, True) self.assertEqual( self.fathers_sister_a in self.member.father.mother.children, True) self.assertEqual( self.fathers_sister_b in self.member.father.mother.children, True) self.assertEqual(self.father in self.member.father.mother.children, True) # test children self.assertEqual(len(self.member.children), 4) self.assertEqual(self.son_a in self.member.children, True) self.assertEqual(self.son_b in self.member.children, True) self.assertEqual(self.daughter_a in self.member.children, True) self.assertEqual(self.daughter_b in self.member.children, True) def test_get_relationship_methods(self): self.assertEqual(len(self.member.get_relationship("paternal_aunt")), 2) self.assertEqual(len(self.member.get_relationship("paternal_uncle")), 2) self.assertEqual(len(self.member.get_relationship("maternal_aunt")), 2) self.assertEqual(len(self.member.get_relationship("maternal_uncle")), 2) self.assertEqual(len(self.member.get_relationship("siblings")), 4) self.assertEqual(len(self.member.get_relationship("son")), 2) self.assertEqual(len(self.member.get_relationship("daughter")), 2) self.assertEqual( len(self.member.spouse.get_relationship("brother_in_law")), 2) self.assertEqual( len(self.member.spouse.get_relationship("sister_in_law")), 2)
class TestMember(TestCase): def setUp(self): self.member = Member(1, "Zim", "Male") def test_initialization(self): # check instance self.assertEqual(isinstance(self.member, Member), True) # check properties self.assertEqual(self.member.id, 1) self.assertEqual(self.member.name, "Zim") self.assertEqual(self.member.gender, Gender.male) self.assertEqual(self.member.mother, None) self.assertEqual(self.member.father, None) self.assertEqual(self.member.spouse, None) self.assertEqual(self.member.children, []) # edge case for gender self.assertRaises(ValueError, Member, 2, "SomeOtherPerson", "Queer") def test_set_mother(self): mother_demo_a = "mother_demo_a" mother_demo_b = Member(2, "MotherDemoB", "Male") mother_demo_c = Member(3, "Mom", "Female") # error case self.assertRaises(ValueError, self.member.set_mother, mother_demo_a) self.assertRaises(ValueError, self.member.set_mother, mother_demo_b) # success case self.member.set_mother(mother_demo_c) self.assertEqual(self.member.mother.name, "Mom") self.assertEqual(self.member.mother.gender, Gender.female) def test_set_father(self): father_demo_a = "family_demo_a" father_demo_b = Member(2, "FatherDemoB", "Female") father_demo_c = Member(3, "Dad", "Male") # error case self.assertRaises(ValueError, self.member.set_father, father_demo_a) self.assertRaises(ValueError, self.member.set_father, father_demo_b) # success case self.member.set_father(father_demo_c) self.assertEqual(self.member.father.name, "Dad") self.assertEqual(self.member.father.gender, Gender.male) def test_set_spouse(self): spouse_demo_a = "spouse_demo_a" spouse_demo_b = Member(2, "SpouseDemoB", "Male") spouse_demo_c = Member(3, "Wife", "Female") # error case self.assertRaises(ValueError, self.member.set_spouse, spouse_demo_a) self.assertRaises(ValueError, self.member.set_spouse, spouse_demo_b) # success case self.member.set_spouse(spouse_demo_c) self.assertEqual(self.member.spouse.name, "Wife") self.assertEqual(self.member.spouse.gender, Gender.female) def test_add_child(self): child_demo_a = "child_demo_a" child_demo_b = Member(4, "Daughter", "Female") # error case self.assertRaises(ValueError, self.member.add_child, child_demo_a) # success case self.member.add_child(child_demo_b) self.assertEqual(len(self.member.children), 1) self.assertEqual(self.member.children[0].name, "Daughter") self.assertEqual(self.member.children[0].gender, Gender.female) def test_get_paternal_grandmother(self): member = Member(9, "Newmember", "Male") father = Member(10, "Newmember_father", "Male") grandmother = Member(11, "Newmember_grandmother", "Female") # error cases self.assertEqual(member.get_paternal_grandmother(), None) member.father = father self.assertEqual(member.get_paternal_grandmother(), None) member.father.mother = grandmother self.assertEqual(member.get_paternal_grandmother(), grandmother) def test_get_maternal_grandmother(self): member = Member(9, "Newmember", "Male") mother = Member(10, "Newmember_mother", "Female") grandmother = Member(11, "Newmember_grandmother", "Female") # error cases self.assertEqual(member.get_paternal_grandmother(), None) member.mother = mother self.assertEqual(member.get_paternal_grandmother(), None) member.mother.mother = grandmother self.assertEqual(member.get_maternal_grandmother(), grandmother) def test_get_spouse_mother(self): member = Member(9, "Newmember", "Male") spouse = Member(10, "Newmember_spouse", "Female") spouse_mother = Member(11, "Newmember_spousemother", "Female") # error cases self.assertEqual(member.get_spouse_mother(), None) member.spouse = spouse self.assertEqual(member.get_spouse_mother(), None) member.spouse.mother = spouse_mother self.assertEqual(member.get_spouse_mother(), spouse_mother) @patch('family_tree.member.Member.get_paternal_grandmother', side_effect=[ None, create_fake_member(), create_fake_member(children=[Member(3, "Dad", "Male")]), create_fake_member(children=[ Member(3, "Dad", "Male"), Member(4, "Uncle", "Male") ]), create_fake_member(children=[ Member(3, "Dad", "Male"), Member(4, "Uncle", "Male"), Member(5, "Aunt", "Female") ]) ]) def test_get_paternal_aunt(self, mock_get_paternal_grandmother): # check if get_paternal_grandmother has been replaced by a mock self.assertEqual( isinstance(self.member.get_paternal_grandmother, Mock), True ) self.assertEqual(self.member.get_paternal_aunt(), []) self.assertEqual(self.member.get_paternal_aunt(), []) self.assertEqual(self.member.get_paternal_aunt(), []) self.assertEqual(self.member.get_paternal_aunt(), []) paternal_aunts = self.member.get_paternal_aunt() self.assertEqual(len(paternal_aunts), 1) self.assertEqual(paternal_aunts[0].name, "Aunt") self.assertEqual(paternal_aunts[0].gender, Gender.female) # to check that the mock_get_paternal_grandmother was called instead # of self.member.get_paternal_grandmother mock_get_paternal_grandmother.assert_called_with() @patch('family_tree.member.Member.get_paternal_grandmother', side_effect=[ None, create_fake_member(), create_fake_member(children=[Member(3, "Dad", "Male")]), create_fake_member(children=[ Member(3, "Aunt", "Female"), Member(4, "Dad", "Male") ]), create_fake_member(children=[ Member(3, "Dad", "Male"), Member(4, "Uncle", "Male"), Member(5, "Aunt", "Female") ]) ]) def test_get_paternal_uncle(self, mock_get_paternal_grandmother): self.member.father = Member(3, "Dad", "Male") # check if get_paternal_grandmother has been replaced by a mock self.assertEqual(isinstance( self.member.get_paternal_grandmother, Mock), True ) self.assertEqual(self.member.get_paternal_uncle(), []) self.assertEqual(self.member.get_paternal_uncle(), []) self.assertEqual(self.member.get_paternal_uncle(), []) self.assertEqual(self.member.get_paternal_uncle(), []) paternal_uncle = self.member.get_paternal_uncle() self.assertEqual(len(paternal_uncle), 1) self.assertEqual(paternal_uncle[0].name, "Uncle") self.assertEqual(paternal_uncle[0].gender, Gender.male) # to check that the mock_get_paternal_grandmother was called instead # of self.member.get_paternal_grandmother mock_get_paternal_grandmother.assert_called_with() @patch('family_tree.member.Member.get_maternal_grandmother', side_effect=[ None, create_fake_member(), create_fake_member(children=[Member(3, "Mom", "Female")]), create_fake_member(children=[ Member(3, "Mom", "Female"), Member(4, "Uncle", "Male") ]), create_fake_member(children=[ Member(3, "Mom", "Female"), Member(4, "Uncle", "Male"), Member(5, "Aunt", "Female") ]) ]) def test_get_maternal_aunt(self, mock_get_maternal_grandmother): self.member.mother = Member(3, "Mom", "Female") # check if get_paternal_grandmother has been replaced by a mock self.assertEqual(isinstance( self.member.get_maternal_grandmother, Mock), True ) self.assertEqual(self.member.get_maternal_aunt(), []) self.assertEqual(self.member.get_maternal_aunt(), []) self.assertEqual(self.member.get_maternal_aunt(), []) self.assertEqual(self.member.get_maternal_aunt(), []) maternal_aunts = self.member.get_maternal_aunt() self.assertEqual(len(maternal_aunts), 1) self.assertEqual(maternal_aunts[0].name, "Aunt") self.assertEqual(maternal_aunts[0].gender, Gender.female) # to check that the mock_get_paternal_grandmother was called instead of # self.member.get_paternal_grandmother mock_get_maternal_grandmother.assert_called_with() @patch('family_tree.member.Member.get_maternal_grandmother', side_effect=[ None, create_fake_member(), create_fake_member(children=[Member(3, "Mom", "Female")]), create_fake_member(children=[ Member(3, "Aunt", "Female"), Member(4, "Mom", "Female") ]), create_fake_member(children=[ Member(3, "Mom", "Female"), Member(4, "Uncle", "Male"), Member(5, "Aunt", "Female") ]) ]) def test_get_maternal_uncle(self, mock_get_maternal_grandmother): # check if get_paternal_grandmother has been replaced by a mock self.assertEqual( isinstance(self.member.get_maternal_grandmother, Mock), True ) self.assertEqual(self.member.get_maternal_uncle(), []) self.assertEqual(self.member.get_maternal_uncle(), []) self.assertEqual(self.member.get_maternal_uncle(), []) self.assertEqual(self.member.get_maternal_uncle(), []) maternal_uncle = self.member.get_maternal_uncle() self.assertEqual(len(maternal_uncle), 1) self.assertEqual(maternal_uncle[0].name, "Uncle") self.assertEqual(maternal_uncle[0].gender, Gender.male) # to check that the mock_get_paternal_grandmother was called # instead of self.member.get_paternal_grandmother mock_get_maternal_grandmother.assert_called_with() @patch('family_tree.member.Member.get_siblings', return_value=[ create_fake_member( name="A", gender=Gender.male, spouse=create_fake_member( name="B", gender=Gender.female, spouse=create_fake_member( name="A") ) ), create_fake_member( name="C", gender=Gender.female, spouse=create_fake_member( name="D", gender=Gender.male, spouse=create_fake_member( name="C") ) ), create_fake_member( name="C", gender=Gender.female ) ]) def test_get_sibling_spouses(self, mock_get_siblings): self.assertEqual(len(self.member.get_sibling_spouses()), 2) def test_get_spouse_siblings(self): self.assertEqual(len(self.member.get_spouse_siblings()), 0) self.member.spouse = create_fake_member(name="Wife") self.member.spouse.get_siblings.return_value = [ create_fake_member(name="A"), create_fake_member(name="B") ] self.assertEqual(len(self.member.get_spouse_siblings()), 2) @patch('family_tree.member.Member.get_spouse_siblings', return_value=[ create_fake_member(name="A", gender=Gender.male), create_fake_member(name="B", gender=Gender.female) ]) @patch('family_tree.member.Member.get_sibling_spouses', return_value=[ create_fake_member(name="C", gender=Gender.male), create_fake_member(name="D", gender=Gender.female) ]) def test_get_brother_in_law(self, mock_get_sibling_spouses, mock_get_spouse_siblings): self.assertEqual(len(self.member.get_brother_in_law()), 2) @patch('family_tree.member.Member.get_spouse_siblings', return_value=[ create_fake_member(name="A", gender=Gender.male), create_fake_member(name="B", gender=Gender.female) ]) @patch('family_tree.member.Member.get_sibling_spouses', return_value=[ create_fake_member(name="C", gender=Gender.male), create_fake_member(name="D", gender=Gender.female) ]) def test_get_sister_in_law(self, mock_get_sibling_spouses, mock_get_spouse_siblings): self.assertEqual(len(self.member.get_sister_in_law()), 2) def test_get_son(self): member = Member(5, "Dummy", "Male") son = Member(7, "Son", "Male") daughter = Member(7, "Daughter", "Female") self.assertEqual(member.get_son(), []) member.children.append(daughter) self.assertEqual(member.get_son(), []) member.children.append(son) sons = member.get_son() self.assertEqual(len(sons), 1) self.assertEqual(sons[0].name, "Son") self.assertEqual(sons[0].gender, Gender.male) def test_get_daughter(self): member = Member(5, "Dummy", "Male") son = Member(7, "Son", "Male") daughter = Member(7, "Daughter", "Female") self.assertEqual(member.get_daughter(), []) member.children.append(son) self.assertEqual(member.get_daughter(), []) member.children.append(daughter) daughters = member.get_daughter() self.assertEqual(len(daughters), 1) self.assertEqual(daughters[0].name, "Daughter") self.assertEqual(daughters[0].gender, Gender.female) def test_get_siblings(self): member = Member(5, "Dummy", "Male") mother = Member(9, "Mother", "Female") son = Member(7, "Son", "Male") daughter = Member(7, "Daughter", "Female") self.assertEqual(member.get_siblings(), []) member.mother = mother self.assertEqual(member.get_siblings(), []) mother.children.extend([member, son, daughter]) member.mother = mother siblings = member.get_siblings() self.assertEqual(len(siblings), 2) @patch('family_tree.member.Member.get_siblings') @patch('family_tree.member.Member.get_daughter') @patch('family_tree.member.Member.get_son') @patch('family_tree.member.Member.get_sister_in_law') @patch('family_tree.member.Member.get_brother_in_law') @patch('family_tree.member.Member.get_maternal_uncle') @patch('family_tree.member.Member.get_maternal_aunt') @patch('family_tree.member.Member.get_paternal_uncle') @patch('family_tree.member.Member.get_paternal_aunt') def test_get_relationship(self, mock_get_paternal_aunt, mock_get_paternal_uncle, mock_get_maternal_aunt, mock_get_maternal_uncle, mock_get_brother_in_law, mock_get_sister_in_law, mock_get_son, mock_get_daughter, mock_get_siblings): self.assertEqual(self.member.get_relationship('invalid_relation'), []) self.member.get_relationship('paternal_aunt') mock_get_paternal_aunt.assert_called_with() self.member.get_relationship('paternal_uncle') mock_get_paternal_uncle.assert_called_with() self.member.get_relationship('maternal_aunt') mock_get_maternal_aunt.assert_called_with() self.member.get_relationship('maternal_uncle') mock_get_maternal_uncle.assert_called_with() self.member.get_relationship('brother_in_law') mock_get_brother_in_law.assert_called_with() self.member.get_relationship('sister_in_law') mock_get_sister_in_law.assert_called_with() self.member.get_relationship('son') mock_get_son.assert_called_with() self.member.get_relationship('daughter') mock_get_daughter.assert_called_with() self.member.get_relationship('siblings') mock_get_siblings.assert_called_with()