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