fix: tests

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 9, 2022 · 2d2cc0c · 2d2cc0c
1 parent 8bff0b7
commit 2d2cc0c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/_internal/io/test_numpy.py b/tests/unit/_internal/io/test_numpy.py
@@ -112,6 +112,8 @@ def test_numpy_openapi_example():
 
 def test_verify_numpy_ndarray(caplog: LogCaptureFixture):
     partial_check = partial(from_example.validate_array, exception_cls=BentoMLException)
+    from_example._enforce_dtype = True
+    from_example._enforce_shape = True
 
     with pytest.raises(BentoMLException) as ex:
         partial_check(np.array(["asdf"]))