p2p-ld/numpydantic
1
0
Fork 0
mirror of https://github.com/p2p-ld/numpydantic.git synced 2024-11-14 18:54:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test shape ellipsis at last

Browse source
This commit is contained in:
sneakers-the-rat 2024-09-23 17:35:24 -07:00
parent f9a992843e
commit 16b0eb0542
Signed by untrusted user who does not match committer: jonny
GPG key ID: 6DCB96EF1E4D232D
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_ndarray.py
View file

@ -172,6 +172,18 @@ def test_ndarray_coercion():
amod = Model(array=["a", "b", "c"]) amod = Model(array=["a", "b", "c"])
@pytest.mark.shape
def test_shape_ellipsis():
"""
Test that ellipsis is a wildcard, rather than "repeat the last index"
"""
class MyModel(BaseModel):
array: NDArray[Shape["1, 2, ..."], Number]
_ = MyModel(array=np.zeros((1, 2, 3, 4, 5)))
@pytest.mark.serialization @pytest.mark.serialization
def test_ndarray_serialize(): def test_ndarray_serialize():
""" """

2
tests/test_serialization.py
View file

@ -12,6 +12,8 @@ from typing import Callable
import numpy as np import numpy as np
import json import json
pytestmark = pytest.mark.serialization
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def hdf5_at_path() -> Callable[[Path], None]: def hdf5_at_path() -> Callable[[Path], None]: