class GraphTestCase(unittest.TestCase):
def setUp(self):
self.graph = TinkerGraph()
def test_create(self):
node1 = {"label": "person", "name": "dave", "age": 34}
resp = self.graph.create(
(node1, "KNOWS", 1), {"label": "lang", "name": "python"})
self.assertEqual(len(resp.vertices), 2)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
out_v = resp.vertices[0]
in_v = resp.vertices[1]
self.assertEqual(edge.source_id, out_v.id)
self.assertEqual(edge.target_id, in_v.id)
def test_create_vertex(self):
resp = self.graph.create(
{"label": "person", "name": "dave", "age": 34})
node1, = resp.vertices
resp = self.graph.create(
(node1, "KNOWS", 1), {"label": "lang", "name": "python"})
self.assertEqual(len(resp.vertices), 1)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
in_v, = resp.vertices
self.assertEqual(edge.target_id, in_v.id)
class GraphTestCase(unittest.TestCase):
def setUp(self):
self.graph = TinkerGraph()
def test_create(self):
node1 = {"label": "person", "name": "dave", "age": 34}
resp = self.graph.create((node1, "KNOWS", 1), {
"label": "lang",
"name": "python"
})
self.assertEqual(len(resp.vertices), 2)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
out_v = resp.vertices[0]
in_v = resp.vertices[1]
self.assertEqual(edge.source_id, out_v.id)
self.assertEqual(edge.target_id, in_v.id)
def test_create_vertex(self):
resp = self.graph.create({
"label": "person",
"name": "dave",
"age": 34
})
node1, = resp.vertices
resp = self.graph.create((node1, "KNOWS", 1), {
"label": "lang",
"name": "python"
})
self.assertEqual(len(resp.vertices), 1)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
in_v, = resp.vertices
self.assertEqual(edge.target_id, in_v.id)
def test_create_multiple(self):
p = {"label": "lang", "name": "python"}
d = {"label": "person", "name": "dave", "age": 34}
f = {"label": "person", "name": "frens", "age": 34}
resp = self.graph.create(
p,
(d, 'KNOWNS', 0),
(f, 'KNOWNS', p),
)
self.assertEqual(len(resp.vertices), 3)
self.assertEqual(len(resp.edges), 2)
e1, e2 = resp.edges
v1, v2, v3 = resp.vertices
self.assertEqual(e1.source_id, v2.id)
self.assertEqual(e1.target_id, v1.id)
self.assertEqual(e2.source_id, v3.id)
self.assertEqual(e2.target_id, v1.id)
class GraphTestCase(unittest.TestCase):
def setUp(self):
self.graph = TinkerGraph()
def test_create(self):
node1 = {"label": "person", "name": "dave", "age": 34}
resp = self.graph.create(
(node1, "KNOWS", 1), {"label": "lang", "name": "python"})
self.assertEqual(len(resp.vertices), 2)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
out_v = resp.vertices[0]
in_v = resp.vertices[1]
self.assertEqual(edge.source_id, out_v.id)
self.assertEqual(edge.target_id, in_v.id)
def test_create_vertex(self):
resp = self.graph.create(
{"label": "person", "name": "dave", "age": 34})
node1, = resp.vertices
resp = self.graph.create(
(node1, "KNOWS", 1), {"label": "lang", "name": "python"})
self.assertEqual(len(resp.vertices), 1)
self.assertEqual(len(resp.edges), 1)
edge, = resp.edges
in_v, = resp.vertices
self.assertEqual(edge.target_id, in_v.id)
def test_create_multiple(self):
p = {"label": "lang", "name": "python"}
d = {"label": "person", "name": "dave", "age": 34}
f = {"label": "person", "name": "frens", "age": 34}
resp = self.graph.create(
p,
(d, 'KNOWNS', 0),
(f, 'KNOWNS', p),
)
self.assertEqual(len(resp.vertices), 3)
self.assertEqual(len(resp.edges), 2)
e1, e2 = resp.edges
v1, v2, v3 = resp.vertices
self.assertEqual(e1.source_id, v2.id)
self.assertEqual(e1.target_id, v1.id)
self.assertEqual(e2.source_id, v3.id)
self.assertEqual(e2.target_id, v1.id)