1#@!#!123s

: / usr / local / lib64 / python3.6 / site-packages / spacy / tests /

Filename : test_architectures.py

back

# coding: utf8
from __future__ import unicode_literals

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")