1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
from rdflib import Graph, URIRef
import unittest
class GraphSlice(unittest.TestCase):
def testSlice(self):
"""
We pervert the slice object,
and use start, stop, step as subject, predicate, object
all operations return generators over full triples
"""
def sl(x, y):
return self.assertEqual(len(list(x)), y)
def soe(x, y):
return self.assertEqual(set([a[2] for a in x]), set(y)) # equals objects
g = self.graph
# Single terms are all trivial:
# single index slices by subject, i.e. return triples((x,None,None))
# tell me everything about "tarek"
sl(g[self.tarek], 2)
# single slice slices by s,p,o, with : used to split
# tell me everything about "tarek" (same as above)
sl(g[self.tarek : :], 2)
# give me every "likes" relationship
sl(g[: self.likes :], 5)
# give me every relationship to pizza
sl(g[:: self.pizza], 3)
# give me everyone who likes pizza
sl(g[: self.likes : self.pizza], 2)
# does tarek like pizza?
self.assertTrue(g[self.tarek : self.likes : self.pizza])
# More intesting is using paths
# everything hated or liked
sl(g[: self.hates | self.likes], 7)
def setUp(self):
self.graph = Graph()
self.michel = URIRef("michel")
self.tarek = URIRef("tarek")
self.bob = URIRef("bob")
self.likes = URIRef("likes")
self.hates = URIRef("hates")
self.pizza = URIRef("pizza")
self.cheese = URIRef("cheese")
self.addStuff()
def addStuff(self):
tarek = self.tarek
michel = self.michel
bob = self.bob
likes = self.likes
hates = self.hates
pizza = self.pizza
cheese = self.cheese
self.graph.add((tarek, likes, pizza))
self.graph.add((tarek, likes, cheese))
self.graph.add((michel, likes, pizza))
self.graph.add((michel, likes, cheese))
self.graph.add((bob, likes, cheese))
self.graph.add((bob, hates, pizza))
self.graph.add((bob, hates, michel)) # gasp!
if __name__ == "__main__":
unittest.main()
|