aboutsummaryrefslogtreecommitdiffstats
path: root/test/extractor/generic/test_path.py
blob: d2b6c61e55c7beebed13076606cbe86a9ba4412a (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
70
71
72
73
74
"""

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.base import extractor
from bsie.utils import bsfs, node as _node, ns

# objects to test
from bsie.extractor.generic.path import Path


## code ##

class TestPath(unittest.TestCase):
    def test_eq(self):
        # distinct instances, same data
        self.assertEqual(Path(), Path())
        # different classes
        class Foo(): pass
        self.assertNotEqual(Path(), Foo())
        self.assertNotEqual(Path(), 123)
        self.assertNotEqual(Path(), None)

    def test_schema(self):
        self.assertEqual(Path().schema,
            bsfs.schema.Schema.from_string(extractor.SCHEMA_PREAMBLE + '''
                bse:filename rdfs:subClassOf bsfs:Predicate ;
                    rdfs:domain bsfs:Entity ;
                    rdfs:range xsd:string ;
                    bsfs:unique "false"^^xsd:boolean .
                '''))

    def test_extract(self):
        ext = Path()
        node = _node.Node(ns.bsfs.File, '') # Blank node
        content = '/tmp/foo/bar'
        p_filename = ext.schema.predicate(ns.bse.filename)
        entity = ext.schema.node(ns.bsfs.Node).get_child(ns.bsfs.Entity)
        string = ext.schema.literal(ns.bsfs.Literal).get_child(ns.xsd.string)

        # baseline
        self.assertSetEqual(set(ext.extract(node, content, (p_filename, ))),
            {(node, p_filename, 'bar')})
        # predicates parameter is respected
        p_foo = ext.schema.predicate(ns.bsfs.Predicate).get_child(ns.bse.foo, domain=entity, range=string) # unsupported predicate
        self.assertSetEqual(set(ext.extract(node, content, (p_filename, p_foo))),
            {(node, p_filename, 'bar')})
        self.assertSetEqual(set(ext.extract(node, content, (p_foo, ))), set())
        # predicates are validated
        p_bar = p_foo.get_child(ns.bse.filename) # same URI but different hierarchy
        self.assertSetEqual(set(ext.extract(node, content, (p_filename, p_bar))),
            {(node, p_filename, 'bar')})
        self.assertSetEqual(set(ext.extract(node, content, (p_bar, ))), set())
        # path variations
        self.assertSetEqual(set(ext.extract(node, 'bar', (p_filename, ))),
            {(node, p_filename, 'bar')})
        self.assertSetEqual(set(ext.extract(node, '', (p_filename, ))),
            {(node, p_filename, '')})
        # errors are suppressed
        self.assertSetEqual(set(ext.extract(node, None, (p_filename, ))), set())


## main ##

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

## EOF ##