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