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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
"""
Part of the bsie test suite.
A copy of the license is provided with the project.
Author: Matthias Baumgartner, 2022
"""
# standard imports
import argparse
import contextlib
import io
import os
import tempfile
import unittest
# external imports
import yaml
# bsie imports
from bsie.utils import bsfs
# objects to test
from bsie.apps.info import main
## code ##
class TestIndex(unittest.TestCase):
def setUp(self):
config = {
'ReaderBuilder': {},
'ExtractorBuilder': [
{'bsie.extractor.preview.Preview': {
'max_sides': [50],
}},
{'bsie.extractor.generic.path.Path': {}},
{'bsie.extractor.generic.constant.Constant': {
'schema': '''
bse:author rdfs:subClassOf bsfs:Predicate ;
rdfs:domain bsfs:Entity ;
rdfs:range xsd:string ;
bsfs:unique "true"^^xsd:boolean .
''',
'tuples': [['http://bsfs.ai/schema/Entity#author', 'Me, myself, and I']],
}},
{'bsie.extractor.image.colors_spatial.ColorsSpatial': {
'width': 2,
'height': 2,
'exp': 2,
}},
]
}
# create config file
_, self.config_path = tempfile.mkstemp(prefix='bsie-test-', suffix='.yaml')
with open(self.config_path, 'wt') as cfile:
yaml.dump(config, cfile)
def tearDown(self):
if os.path.exists(self.config_path):
os.unlink(self.config_path)
def test_predicates(self):
outbuf = io.StringIO()
with contextlib.redirect_stdout(outbuf):
# show predicates infos
main(['--config', self.config_path, 'predicates'])
# verify output
self.assertSetEqual({pred for pred in outbuf.getvalue().split('\n') if pred != ''}, {
'http://bsfs.ai/schema/Entity#author',
'http://bsfs.ai/schema/Predicate',
'http://bsfs.ai/schema/Entity#filename',
'http://bsfs.ai/schema/Entity/colors_spatial#0658f2234a054e1dd59a14462c89f7733e019160419c796356aa831498bd0a04',
'http://bsfs.ai/schema/Entity#preview',
'http://bsfs.ai/schema/Preview#width',
'http://bsfs.ai/schema/Preview#height',
'http://bsfs.ai/schema/Preview#asset',
})
def test_schema(self):
outbuf = io.StringIO()
with contextlib.redirect_stdout(outbuf):
# show schema infos
main(['--config', self.config_path, 'schema'])
# verify output
schema = bsfs.schema.from_string(outbuf.getvalue())
self.assertSetEqual({pred.uri for pred in schema.predicates()}, {
'http://bsfs.ai/schema/Entity#author',
'http://bsfs.ai/schema/Predicate',
'http://bsfs.ai/schema/Entity#filename',
'http://bsfs.ai/schema/Entity/colors_spatial#0658f2234a054e1dd59a14462c89f7733e019160419c796356aa831498bd0a04',
'http://bsfs.ai/schema/Entity#preview',
'http://bsfs.ai/schema/Preview#width',
'http://bsfs.ai/schema/Preview#height',
'http://bsfs.ai/schema/Preview#asset',
})
def test_invalid(self):
with contextlib.redirect_stderr(io.StringIO()):
self.assertRaises(SystemExit, main, ['foobar'])
## main ##
if __name__ == '__main__':
unittest.main()
## EOF ##
|
