@@ -0,0 +1,250 @@
+import pytest%0Aimport numpy%0A%0Afrom ...neural._classes.model import Model%0A%0A%0A%0Adef test_mismatched_shapes_raise_ShapeError():%0A    X = numpy.ones((3, 4))%0A    model = Model(10, 5)%0A    with pytest.raises(ValueError):%0A        y = model.begin_training(X)%0A    %0A

Add shape check