def test_subgraph_edge_count_grandparent(self):
relative = fake.family_member(sex='Female')
a1 = Person(first_name=relative['first_name'],
last_name=relative['last_name'],
ethnic_name=relative['ethnic_name'],
sex=relative['sex'],
birth_date=relative['birth_date'],
email=relative['email'],
confirmed=False)
relative = fake.family_member(sex='Male')
a2 = Person(first_name=relative['first_name'],
last_name=relative['last_name'],
ethnic_name=relative['ethnic_name'],
sex=relative['sex'],
birth_date=relative['birth_date'],
email=relative['email'],
confirmed=False)
db.session.add_all([a1, a2])
db.session.commit()
self.p3.get_or_create_relationship(a1, 1)
self.p3.get_or_create_relationship(a2, 3)
a1.get_or_create_relationship(a2, 3)
n1_subgraph = self.p1.get_graph()
c6 = graph.count_relationship_weights(self.p1, n1_subgraph)
self.assertEqual(c6.get(1), 2)
self.assertIsNone(c6.get(2))
self.assertEqual(c6.get(3), 3)
self.assertIsNone(c6.get(4))
