Пример #1
0
    def test_edge_label_string_traversal_types(self):
        """ Test traversals with edge instances work properly """
        e1 = TestEdgeModel.create(self.v1, self.v2)
        e2 = OtherTestEdge.create(self.v1, self.v3)
        e3 = YetAnotherTestEdge.create(self.v1, self.v4)

        out = self.v1.outV(TestEdgeModel.get_label(), OtherTestEdge.get_label())
        self.assertEqual(len(out), 2)
        self.assertIn(self.v2.id, [v.id for v in out])
        self.assertIn(self.v3.id, [v.id for v in out])

        out = self.v1.outV(OtherTestEdge.get_label(), YetAnotherTestEdge.get_label())
        self.assertEqual(len(out), 2)
        self.assertIn(self.v3.id, [v.id for v in out])
        self.assertIn(self.v4.id, [v.id for v in out])

        e1.delete()
        e2.delete()
        e3.delete()