x == y or z == y → y in (x, z)

Suggested by refurb.

bench/ndarray/download_data.py

@@ -15,7 +15,7 @@ def open_zarr(year, month, datestart, dateend, dset):
     datestring = "era5-pds/zarr/{year}/{month:02d}/data/".format(year=year, month=month)
     s3map = s3fs.S3Map(datestring + dset + ".zarr/", s3=fs)
     arr = xr.open_dataset(s3map, engine="zarr")
-    if dset[:3] == "air" or dset[:3] == "sno" or dset[:3] == "eas":
+    if dset[:3] in ("air", "sno", "eas"):
         arr = arr.sel(time0=slice(np.datetime64(datestart), np.datetime64(dateend)))
     else:
         arr = arr.sel(time1=slice(np.datetime64(datestart), np.datetime64(dateend)))

tests/ndarray/test_empty.py

@@ -104,7 +104,7 @@ def test_empty_minimal(shape, dtype):
     a = blosc2.empty(shape, dtype=dtype)
 
     dtype = np.dtype(dtype)
-    assert a.shape == shape or a.shape[0] == shape
+    assert shape in (a.shape, a.shape[0])
     assert a.chunks is not None
     assert a.blocks is not None
     assert all(c >= b for c, b in zip(a.chunks, a.blocks))

tests/ndarray/test_struct_dtype.py

@@ -30,7 +30,7 @@ def test_scalar(shape, dtype, urlpath):
     assert np.array_equal(a[:], b)
 
     dtype = np.dtype(dtype)
-    assert a.shape == shape or a.shape[0] == shape
+    assert shape in (a.shape, a.shape[0])
     assert a.dtype == dtype
     assert a.schunk.typesize == dtype.itemsize
     assert a.shape == b.shape

tests/ndarray/test_zeros.py

@@ -114,7 +114,7 @@ def test_zeros_minimal(shape, dtype):
     assert np.array_equal(a[:], b)
 
     dtype = np.dtype(dtype)
-    assert a.shape == shape or a.shape[0] == shape
+    assert shape in (a.shape, a.shape[0])
     assert a.chunks is not None
     assert a.blocks is not None
     assert all(c >= b for c, b in zip(a.chunks, a.blocks))