def testVertexOut(self):
"""
Vertex().out() should ret a list of the vertices that are 'out' from it
"""
u = Vertex()
v = Vertex()
e = u >> v
self.assertEqual(u.out(), {v})
def testAttributeSelectorWithFilterMultipleResults(self):
"""Make sure having multiple results doesn't mess it up"""
t = Vertex(name="_t", value=1)
u = Vertex(name="_u", value=5)
v = Vertex(name="_v", value=5)
e1 = t >> u
e2 = t >> v
result = t.out(value=5).name
self.assertEqual(result, {"_u", "_v"})
def testAttributeSelectorWithFilter(self):
"""Make sure you can still get the attribute after using a filter"""
t = Vertex(name="_t", value=1)
u = Vertex(name="_u", value=3)
v = Vertex(name="_v", value=5)
e1 = t >> u
e2 = t >> v
result = t.out(value=5).name
self.assertEqual(result, {"_v"})
def testMultipleFiltersMultipleResultsSomeMissingAttribute(self):
"""Ensure multiple filters get applied correctly"""
t = Vertex()
u = Vertex(name="_u", value=3)
v = Vertex(name="_v", class_="blah", value=4)
w = Vertex(name="_w", class_="blah", value=4)
e1 = t >> u
e2 = t >> v
e3 = t >> w
result = t.out(class_="blah", value=4)
self.assertEqual(result, {v, w})
def testSelectorWithFilterMultipleResults(self):
"""
Ensure a query with multiple results gets returned correctly
"""
t = Vertex(name="_t", value=1)
u = Vertex(name="_u", value=5)
v = Vertex(name="_v", value=5)
e1 = t >> u
e2 = t >> v
result = t.out(value=5)
self.assertEqual(result, {u, v})
def testSelectorWithFilter(self):
"""
Selectors take optional keyword arguments as filters
"""
t = Vertex(name="_t", value=1)
u = Vertex(name="_u", value=3)
v = Vertex(name="_v", value=5)
e1 = t >> u
e2 = t >> v
result = t.out(value=5)
self.assertEqual(result, {v})
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"})
