1
0
mirror of https://github.com/ilri/csv-metadata-quality.git synced 2024-11-26 07:38:18 +01:00

tests/test_check.py: add tests for dropping invalid AGROVOC

This commit is contained in:
Alan Orth 2021-12-23 12:44:32 +02:00
parent a7727b8431
commit c43095139a
Signed by: alanorth
GPG Key ID: 0FB860CC9C45B1B9

View File

@ -179,18 +179,41 @@ def test_check_invalid_language(capsys):
def test_check_invalid_agrovoc(capsys): def test_check_invalid_agrovoc(capsys):
"""Test invalid AGROVOC subject.""" """Test invalid AGROVOC subject. Invalid values *will not* be dropped."""
value = "FOREST" valid_agrovoc = "LIVESTOCK"
invalid_agrovoc = "FOREST"
value = f"{valid_agrovoc}||{invalid_agrovoc}"
field_name = "dcterms.subject" field_name = "dcterms.subject"
drop = False
check.agrovoc(value, field_name) new_value = check.agrovoc(value, field_name, drop)
captured = capsys.readouterr() captured = capsys.readouterr()
assert ( assert (
captured.out captured.out
== f"{Fore.RED}Invalid AGROVOC ({field_name}): {Fore.RESET}{value}\n" == f"{Fore.RED}Invalid AGROVOC ({field_name}): {Fore.RESET}{invalid_agrovoc}\n"
) )
assert new_value == value
def test_check_invalid_agrovoc_dropped(capsys):
"""Test invalid AGROVOC subjects. Invalid values *will* be dropped."""
valid_agrovoc = "LIVESTOCK"
invalid_agrovoc = "FOREST"
value = f"{valid_agrovoc}||{invalid_agrovoc}"
field_name = "dcterms.subject"
drop = True
new_value = check.agrovoc(value, field_name, drop)
captured = capsys.readouterr()
assert (
captured.out
== f"{Fore.GREEN}Dropping invalid AGROVOC ({field_name}): {Fore.RESET}{invalid_agrovoc}\n"
)
assert new_value == valid_agrovoc
def test_check_valid_agrovoc(): def test_check_valid_agrovoc():
@ -198,10 +221,11 @@ def test_check_valid_agrovoc():
value = "FORESTS" value = "FORESTS"
field_name = "dcterms.subject" field_name = "dcterms.subject"
drop = False
result = check.agrovoc(value, field_name) result = check.agrovoc(value, field_name, drop)
assert result == None assert result == "FORESTS"
def test_check_uncommon_filename_extension(capsys): def test_check_uncommon_filename_extension(capsys):