def test_one_child_child(self): p1 = mock.create_autospec(Person, idn='I1', family_connections=['F1']) p2 = mock.create_autospec(Person, idn='I2', family_connections=['F1']) f1 = mock.create_autospec(Family, idn='F1', head='I1', partner='I2', family_connections=['I3']) p3 = mock.create_autospec(Person, idn='I3', origin='F1', family_connections=None) e = Env({p1.idn: p1, p2.idn: p2, p3.idn: p3, f1.idn: f1}) self.assertEqual(e.get_children(p3), set())
def test_no_children(self): p1 = mock.create_autospec(Person, idn='I1', family_connections=None) e = Env({p1.idn: p1}) self.assertEqual(e.get_children(p1), set())