class TitanGraphTestCase(unittest.TestCase):
def setUp(self):
self.graph = TitanGraph()
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 setUp(self):
self.graph = TitanGraph()