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())