def test_append_edge_with_ident_set(self):
node1 = Vertex(label="NODE")
node2 = Vertex(label="NODE")
edge = Edge(node1, "knows", node2)
edge.ident = 0
self.assertRaises(interfaces.DatabaseException, self.graph.append_edge,
edge)
def test_union_filter_for_labels_with_no_props(self):
id_generator = IDGenerator()
marko = Vertex(
"person",
name="Marko",
)
josh = Vertex(
"person",
name="Marko",
)
edge = Edge(marko, "father", josh)
edge2 = Edge(josh, "son", marko)
# attach ID's to the vertices and edges.
marko.ident = id_generator.get_vertex_id()
josh.ident = id_generator.get_vertex_id()
edge.ident = id_generator.get_edge_id()
edge2.ident = id_generator.get_edge_id()
container1 = EntitySet()
container2 = EntitySet()
container1.add(edge)
container2.add(edge2)
container3 = container1 | container2
self.assertEqual(
container3.filter("father").sorted(),
sorted([edge])
)
self.assertEqual(
container3.filter("son").sorted(),
sorted([edge2])
)
def test_is_bound(self):
edge = Edge(self.marko, "de-friends", self.josh)
graph = Graph()
graph.bind_to_graph(edge)
self.assertEqual(
edge.is_bound(),
True
)
def test_append_edge(self):
node1 = Vertex(label="NODE")
node2 = Vertex(label="NODE")
edge = Edge(node1, "knows", node2)
self.graph.append_edge(edge)
self.assertEqual(edge.ident, self.graph._id_tracker.eid - 1)
self.assertIn(edge, self.graph)
self.assertEqual(edge.is_bound(), True)
def test_append_edge_with_ident_set(self):
node1 = Vertex(label="NODE")
node2 = Vertex(label="NODE")
edge = Edge(node1, "knows", node2)
edge.ident = 0
self.assertRaises(
interfaces.DatabaseException,
self.graph.append_edge,
edge
)
def test_union_filter_for_labels_with_no_props(self):
id_generator = IDGenerator()
marko = Vertex(
"person",
name="Marko",
)
josh = Vertex(
"person",
name="Marko",
)
edge = Edge(marko, "father", josh)
edge2 = Edge(josh, "son", marko)
# attach ID's to the vertices and edges.
marko.ident = id_generator.get_vertex_id()
josh.ident = id_generator.get_vertex_id()
edge.ident = id_generator.get_edge_id()
edge2.ident = id_generator.get_edge_id()
container1 = EntitySet()
container2 = EntitySet()
container1.add(edge)
container2.add(edge2)
container3 = container1 | container2
self.assertEqual(container3.filter("father").sorted(), sorted([edge]))
self.assertEqual(container3.filter("son").sorted(), sorted([edge2]))
def test_filter_for_labels_with_no_props(self):
marko = Vertex(
"person",
name="Marko",
)
josh = Vertex(
"person",
name="Marko",
)
edge = Edge(marko, "father", josh)
container = EntitySet()
container.add(edge)
self.assertEqual(container.filter("father").sorted(), sorted([edge]))
def test_is_bound_not_bound(self):
edge = Edge(self.marko, "de-friends", self.josh)
self.assertEqual(
edge.is_bound(),
False
)
def test_contains_edge_not_found(self):
e = Edge(self.marko, "friend", self.josh)
self.assertEqual(self.graph.__contains__(e), False)
