Skip to content

Commit 4a7a550

Browse files
committed
docstrings and simplification for regression tests
1 parent 39219fa commit 4a7a550

File tree

1 file changed

+7
-19
lines changed

1 file changed

+7
-19
lines changed

tests/test_regression/test_regression.py

Lines changed: 7 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import subprocess
2-
from dataclasses import asdict, dataclass
2+
from dataclasses import dataclass
33
from itertools import product
44
from pathlib import Path
55

@@ -11,36 +11,24 @@
1111
import zarr
1212
from zarr.core.array import Array
1313
from zarr.core.dtype.npy.string import VariableLengthString
14-
from zarr.core.metadata.v2 import ArrayV2Metadata
1514
from zarr.storage import LocalStore
1615

1716

1817
def runner_installed() -> bool:
18+
"""
19+
Check if a PEP-723 compliant python script runner is installed.
20+
"""
1921
try:
2022
subprocess.check_output(["uv", "--version"])
21-
return True
23+
return True # noqa: TRY300
2224
except FileNotFoundError:
2325
return False
2426

2527

26-
def array_metadata_equals(a: ArrayV2Metadata, b: ArrayV2Metadata) -> bool:
27-
dict_a, dict_b = asdict(a), asdict(b)
28-
fill_value_a, fill_value_b = dict_a.pop("fill_value"), dict_b.pop("fill_value")
29-
if (
30-
isinstance(fill_value_a, float)
31-
and isinstance(fill_value_b, float)
32-
and np.isnan(fill_value_a)
33-
and np.isnan(fill_value_b)
34-
):
35-
return dict_a == dict_b
36-
else:
37-
return fill_value_a == fill_value_b and dict_a == dict_b
38-
39-
4028
@dataclass(kw_only=True)
4129
class ArrayParams:
4230
values: np.ndarray[tuple[int], np.dtype[np.generic]]
43-
fill_value: np.generic | str
31+
fill_value: np.generic | str | int
4432
compressor: numcodecs.abc.Codec
4533

4634

@@ -121,5 +109,5 @@ def test_roundtrip(source_array: Array, tmp_path: Path) -> None:
121109
)
122110
assert copy_op.returncode == 0
123111
out_array = zarr.open_array(store=out_path, mode="r", zarr_format=2)
124-
assert array_metadata_equals(source_array.metadata, out_array.metadata)
112+
assert source_array.metadata.to_dict() == out_array.metadata.to_dict()
125113
assert np.array_equal(source_array[:], out_array[:])

0 commit comments

Comments
 (0)