def test_all_full(self): """ Test RDFQueryReader.all() method with full=True. """ concept = ns.FOAF["PERSON"] expected_result = [("a", "b", "c")] query_reader = RDFQueryReader(use_subqueries = True) def triples_func(*args, **kwargs): return expected_result query_reader._triples = triples_func result = query_reader._all(concept, full = True) self.assertEqual(result, expected_result)
def test_all_subjects(self): """ Test RDFQueryReader.all() method. Mock query_reader._values function, test that _all returns list of URIs. """ concept = ns.FOAF["PERSON"] expected_result = {"uri1" : [concept], "uri2" : [concept]} query_reader = RDFQueryReader() def values_func(*args, **kwargs): return expected_result query_reader._values = values_func result = query_reader._all(concept) self.assertEqual(set(result), set(["uri1", "uri2"]))