import unittest from abc import ABC, abstractmethod from io import BytesIO from pyoxigraph import * foo = NamedNode("http://foo") bar = NamedNode("http://bar") baz = NamedNode("http://baz") graph = NamedNode("http://graph") class TestAbstractStore(unittest.TestCase, ABC): @abstractmethod def store(self): pass def test_add(self): store = self.store() store.add(Quad(foo, bar, baz)) store.add(Quad(foo, bar, baz, DefaultGraph())) store.add(Quad(foo, bar, baz, graph)) self.assertEqual(len(store), 2) def test_remove(self): store = self.store() store.add(Quad(foo, bar, baz)) store.add(Quad(foo, bar, baz, DefaultGraph())) store.add(Quad(foo, bar, baz, graph)) store.remove(Quad(foo, bar, baz)) self.assertEqual(len(store), 1) def test_len(self): store = self.store() store.add(Quad(foo, bar, baz)) store.add(Quad(foo, bar, baz, graph)) self.assertEqual(len(store), 2) def test_in(self): store = self.store() store.add(Quad(foo, bar, baz)) store.add(Quad(foo, bar, baz, DefaultGraph())) store.add(Quad(foo, bar, baz, graph)) self.assertTrue(Quad(foo, bar, baz) in store) self.assertTrue(Quad(foo, bar, baz, DefaultGraph()) in store) self.assertTrue(Quad(foo, bar, baz, graph) in store) self.assertTrue(Quad(foo, bar, baz, foo) not in store) def test_iter(self): store = self.store() store.add(Quad(foo, bar, baz, DefaultGraph())) store.add(Quad(foo, bar, baz, graph)) self.assertEqual( set(store), {Quad(foo, bar, baz, DefaultGraph()), Quad(foo, bar, baz, graph)}, ) def test_quads_for_pattern(self): store = self.store() store.add(Quad(foo, bar, baz, DefaultGraph())) store.add(Quad(foo, bar, baz, graph)) self.assertEqual( set(store.quads_for_pattern(None, None, None)), {Quad(foo, bar, baz, DefaultGraph()), Quad(foo, bar, baz, graph)}, ) self.assertEqual( set(store.quads_for_pattern(foo, None, None)), {Quad(foo, bar, baz, DefaultGraph()), Quad(foo, bar, baz, graph)}, ) self.assertEqual( set(store.quads_for_pattern(None, None, None, graph)), {Quad(foo, bar, baz, graph)}, ) self.assertEqual( set(store.quads_for_pattern(foo, None, None, DefaultGraph())), {Quad(foo, bar, baz, DefaultGraph())}, ) def test_ask_query(self): store = self.store() store.add(Quad(foo, foo, foo)) self.assertTrue(store.query("ASK { ?s ?s ?s }")) self.assertFalse(store.query("ASK { FILTER(false) }")) def test_construct_query(self): store = self.store() store.add(Quad(foo, bar, baz)) self.assertEqual( set(store.query("CONSTRUCT { ?s ?p ?o } WHERE { ?s ?p ?o }")), {Triple(foo, bar, baz)}, ) def test_select_query(self): store = self.store() store.add(Quad(foo, bar, baz)) results = list(store.query("SELECT ?s WHERE { ?s ?p ?o }")) self.assertEqual(len(results), 1) self.assertEqual(results[0][0], foo) self.assertEqual(results[0]["s"], foo) def test_load_ntriples_to_default_graph(self): store = self.store() store.load( BytesIO(b" ."), mime_type="application/n-triples", ) self.assertEqual(set(store), {Quad(foo, bar, baz, DefaultGraph())}) def test_load_ntriples_to_named_graph(self): store = self.store() store.load( BytesIO(b" ."), mime_type="application/n-triples", to_graph=graph, ) self.assertEqual(set(store), {Quad(foo, bar, baz, graph)}) def test_load_turtle_with_base_iri(self): store = self.store() store.load( BytesIO(b" <> ."), mime_type="text/turtle", base_iri="http://baz", ) self.assertEqual(set(store), {Quad(foo, bar, baz, DefaultGraph())}) def test_load_nquads(self): store = self.store() store.load( BytesIO(b" ."), mime_type="application/n-quads", ) self.assertEqual(set(store), {Quad(foo, bar, baz, graph)}) def test_load_trig_with_base_iri(self): store = self.store() store.load( BytesIO(b" { <> . }"), mime_type="application/trig", base_iri="http://baz", ) self.assertEqual(set(store), {Quad(foo, bar, baz, graph)}) def test_dump_ntriples(self): store = self.store() store.add(Quad(foo, bar, baz, graph)) output = BytesIO() store.dump(output, "application/n-triples", from_graph=graph) self.assertEqual( output.getvalue(), b" .\n", ) def test_dump_nquads(self): store = self.store() store.add(Quad(foo, bar, baz, graph)) output = BytesIO() store.dump(output, "application/n-quads") self.assertEqual( output.getvalue(), b" .\n", ) class TestMemoryStore(TestAbstractStore): def store(self): return MemoryStore() class TestSledStore(TestAbstractStore): def store(self): return SledStore() del TestAbstractStore # We do not want to expose this class to the test runner if __name__ == "__main__": unittest.main()