spaCy/spacy/tests/morphology/test_morph_features.py
Daniël de Kok e2b70df012
Configure isort to use the Black profile, recursively isort the spacy module (#12721)
* Use isort with Black profile

* isort all the things

* Fix import cycles as a result of import sorting

* Add DOCBIN_ALL_ATTRS type definition

* Add isort to requirements

* Remove isort from build dependencies check

* Typo
2023-06-14 17:48:41 +02:00

51 lines
1.3 KiB
Python

import pytest
from spacy.morphology import Morphology
from spacy.strings import StringStore, get_string_id
@pytest.fixture
def morphology():
return Morphology(StringStore())
def test_init(morphology):
pass
def test_add_morphology_with_string_names(morphology):
morphology.add({"Case": "gen", "Number": "sing"})
def test_add_morphology_with_int_ids(morphology):
morphology.strings.add("Case")
morphology.strings.add("gen")
morphology.strings.add("Number")
morphology.strings.add("sing")
morphology.add(
{
get_string_id("Case"): get_string_id("gen"),
get_string_id("Number"): get_string_id("sing"),
}
)
def test_add_morphology_with_mix_strings_and_ints(morphology):
morphology.strings.add("PunctSide")
morphology.strings.add("ini")
morphology.add(
{get_string_id("PunctSide"): get_string_id("ini"), "VerbType": "aux"}
)
def test_morphology_tags_hash_distinctly(morphology):
tag1 = morphology.add({"PunctSide": "ini", "VerbType": "aux"})
tag2 = morphology.add({"Case": "gen", "Number": "sing"})
assert tag1 != tag2
def test_morphology_tags_hash_independent_of_order(morphology):
tag1 = morphology.add({"Case": "gen", "Number": "sing"})
tag2 = morphology.add({"Number": "sing", "Case": "gen"})
assert tag1 == tag2