mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-13 18:56:36 +03:00
db55577c45
* Remove unicode declarations * Remove Python 3.5 and 2.7 from CI * Don't require pathlib * Replace compat helpers * Remove OrderedDict * Use f-strings * Set Cython compiler language level * Fix typo * Re-add OrderedDict for Table * Update setup.cfg * Revert CONTRIBUTING.md * Revert lookups.md * Revert top-level.md * Small adjustments and docs [ci skip]
17 lines
445 B
Python
17 lines
445 B
Python
import pytest
|
|
from spacy import registry
|
|
from thinc.v2v import Affine
|
|
from catalogue import RegistryError
|
|
|
|
|
|
@registry.architectures.register("my_test_function")
|
|
def create_model(nr_in, nr_out):
|
|
return Affine(nr_in, nr_out)
|
|
|
|
|
|
def test_get_architecture():
|
|
arch = registry.architectures.get("my_test_function")
|
|
assert arch is create_model
|
|
with pytest.raises(RegistryError):
|
|
registry.architectures.get("not_an_existing_key")
|