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