spaCy/spacy/tests/morphology/test_morph_features.py
2023-06-26 11:41:03 +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