1#@!#!123s

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

Filename : test_shape_check.py

back

# coding: utf8
from __future__ import unicode_literals

import pytest
import numpy

from ...neural._classes.affine import Affine
from ...exceptions import ShapeMismatchError


def test_mismatched_shapes_raise_ShapeMismatchError():
    X = numpy.ones((3, 4))
    model = Affine(10, 5)
    with pytest.raises(ShapeMismatchError):
        model.predict(X)