# -*- coding: utf-8 -*-
from anchorman import annotate, get_config
from anchorman.generator.candidate import attributes_of
from anchorman.configuration import parse_yaml
DATA = parse_yaml('data.yaml', loaded_from=__file__)
def my_validator(item, candidates, this_unit, settings):
"""Write your own validator based on this values.
You can write your own validator function, because you get access to the
following values. Pass values via settings if needed.
Args:
item (Interval): Interval(65, 68, ('dog', ('entity', {'dog': {'score': 12.0, 'type': 'animal', 'value': '/wiki/dog'}})))
candidates (list): [Interval(21, 27, ('q\xc3\xbcick', ('entity', {'q\xc3\xbcick': {'score': 0.2, 'type': 'term', 'value': '/wiki/queick'}}))), Interval(34, 37, ('fox', ('entity', {'fox': {'score': 23.0, 'type': 'animal', 'value': '/wiki/fox'}}))), Interval(60, 64, ('lazy', ('entity', {'lazy': {'score': 5.55, 'type': 'term', 'value': '/wiki/lazy'}})))]
this_unit (list): [Interval(60, 64, ('lazy', ('entity', {'lazy': {'score': 5.55, 'type': 'term', 'value': '/wiki/lazy'}})))]
settings (dict): {'mode': 'highlight', 'element_identifier': 'entity', 'text_unit': {'name': 'html-paragraph', 'key': 'p', 'number_of_items': None}, 'longest_match_first': True, 'replaces_at_all': None, 'case_sensitive': True}
Returns:
bool: True if element is valid candidate, False otherwise.
"""
attributes = attributes_of(item)
_score = attributes['score']
_type = attributes['type']
return True if _score >= 42.0 and _type == 'city' else False
def test_yaml_parsing_error():
with pytest.raises(YAMLError):
_ = parse_yaml('data_error.yaml', loaded_from=__file__)
def test_parse_yaml():
DATA = parse_yaml('data.yaml', loaded_from=__file__)
assert isinstance(DATA, dict)
def test_yaml_io_error():
with pytest.raises(IOError):
_ = parse_yaml('nofilelikethis.yaml')
