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"})
