aboutsummaryrefslogtreecommitdiffstats
path: root/test/base/test_extractor.py
blob: 5410ae07db5c99df73b2fa719069204905518f3b (plain)
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
"""

Part of the bsie test suite.
A copy of the license is provided with the project.
Author: Matthias Baumgartner, 2022
"""
# imports
import unittest

# bsie imports
from bsie.utils import bsfs, ns

# objects to test
from bsie.base import extractor


## code ##

class StubExtractor(extractor.Extractor):
    def __init__(self):
        super().__init__(bsfs.schema.Schema.from_string(extractor.SCHEMA_PREAMBLE + '''
            bse:author rdfs:subClassOf bsfs:Predicate ;
                rdfs:domain bsfs:Entity ;
                rdfs:range xsd:string ;
                bsfs:unique "false"^^xsd:boolean .
            bse:comment rdfs:subClassOf bsfs:Predicate ;
                rdfs:domain bsfs:Entity ;
                rdfs:range xsd:string ;
                bsfs:unique "false"^^xsd:boolean .
        '''))

    def extract(self, subject, content, predicates):
        raise NotImplementedError()

class StubSub(StubExtractor):
    pass

class TestExtractor(unittest.TestCase):
    def test_essentials(self):
        ext = StubExtractor()
        self.assertEqual(str(ext), 'StubExtractor')
        self.assertEqual(repr(ext), 'StubExtractor()')
        self.assertEqual(ext, StubExtractor())
        self.assertEqual(hash(ext), hash(StubExtractor()))

        sub = StubSub()
        self.assertEqual(str(sub), 'StubSub')
        self.assertEqual(repr(sub), 'StubSub()')
        self.assertEqual(sub, StubSub())
        self.assertEqual(hash(sub), hash(StubSub()))
        self.assertNotEqual(ext, sub)
        self.assertNotEqual(hash(ext), hash(sub))

    def test_predicates(self):
        schema = bsfs.schema.Schema.Empty()
        entity = schema.node(ns.bsfs.Node).get_child(ns.bsfs.Entity)
        string = schema.literal(ns.bsfs.Literal).get_child(bsfs.URI('http://www.w3.org/2001/XMLSchema#string'))
        p_author = schema.predicate(ns.bsfs.Predicate).get_child(ns.bse.author, domain=entity, range=string)
        p_comment = schema.predicate(ns.bsfs.Predicate).get_child(ns.bse.comment, domain=entity, range=string)
        ext = StubExtractor()
        self.assertSetEqual(set(ext.predicates()), {p_author, p_comment} | set(schema.predicates()))


## main ##

if __name__ == '__main__':
    unittest.main()

## EOF ##