def testChainingSelectorsWithAttributeMultiple(self): """ Chain selectors together, and get a list of attributes """ t = Vertex() u = Vertex() v = Vertex() t.name = "_t" u.name = "_u" v.name = "_v" e1 = t >> v e2 = u >> v result = t.out().in_().name self.assertEqual(result, {"_t", "_u"})
def testChainingSelectorsWithAttributeAccess(self): """ Chain selectors together and request an attribute at the end of the chain """ t = Vertex() u = Vertex() u.name = "I am U" e = t >> u should_be_i_am_u = t.outE().inV().name self.assertEqual(should_be_i_am_u, {"I am U"})