def test_should_fail_on_bad_node_type(self): store = FrozenGraphStore(self.store) with self.assertRaises(TypeError): _ = store.relationships(n_ids=1)
def test_should_fail_on_bad_node_sequence(self): store = FrozenGraphStore(self.store) self.assertEqual( list(store.relationships(n_ids=(self.a, self.b, self.c))), [])
def test_should_fail_on_bad_node_set(self): store = FrozenGraphStore(self.store) _ = store.relationships(n_ids={self.a, self.b, self.c})
def test_should_get_relationships(self): store = FrozenGraphStore(self.store) self.assertEqual( set(store.relationships()), { self.a_likes_b, self.b_likes_a, self.a_knows_b, self.a_knows_c, self.c_knows_b, self.c_married_to_d }) self.assertEqual(set(store.relationships("KNOWS")), {self.a_knows_b, self.a_knows_c, self.c_knows_b}) self.assertEqual(set(store.relationships("MARRIED_TO")), {self.c_married_to_d}) self.assertEqual(set(store.relationships(n_ids=(self.a, None))), {self.a_likes_b, self.a_knows_b, self.a_knows_c}) self.assertEqual(set(store.relationships("KNOWS", (self.a, None))), {self.a_knows_b, self.a_knows_c}) self.assertEqual(set(store.relationships(n_ids=(None, self.b))), {self.a_likes_b, self.a_knows_b, self.c_knows_b}) self.assertEqual( set(store.relationships("KNOWS", n_ids=(None, self.b))), {self.a_knows_b, self.c_knows_b}) self.assertEqual(set(store.relationships(n_ids=(self.a, self.b))), {self.a_likes_b, self.a_knows_b}) self.assertEqual(set(store.relationships("KNOWS", (self.a, self.b))), {self.a_knows_b}) self.assertEqual( set(store.relationships(n_ids={self.a})), {self.a_likes_b, self.b_likes_a, self.a_knows_b, self.a_knows_c}) self.assertEqual(set(store.relationships("KNOWS", {self.a})), {self.a_knows_b, self.a_knows_c}) self.assertEqual(set(store.relationships(n_ids={self.a, self.b})), {self.a_likes_b, self.b_likes_a, self.a_knows_b}) self.assertEqual( set(store.relationships("KNOWS", n_ids={self.a, self.b})), {self.a_knows_b})