Add more setitem property tests

Author: dcherian

tests/test_properties.py

@@ -1,12 +1,11 @@
-import numpy as np
 import pytest
 from numpy.testing import assert_array_equal
 
 pytest.importorskip("hypothesis")
 
 import hypothesis.extra.numpy as npst
 import hypothesis.strategies as st
-from hypothesis import given
+from hypothesis import assume, given
 
 from zarr.testing.strategies import (
     arrays,
@@ -32,7 +31,7 @@ def test_basic_indexing(data: st.DataObject) -> None:
     actual = zarray[indexer]
     assert_array_equal(nparray[indexer], actual)
 
-    new_data = np.ones_like(actual)
+    new_data = data.draw(npst.arrays(shape=st.just(actual.shape), dtype=nparray.dtype))
     zarray[indexer] = new_data
     nparray[indexer] = new_data
     assert_array_equal(nparray, zarray[:])
@@ -48,6 +47,11 @@ def test_oindex(data: st.DataObject) -> None:
     actual = zarray.oindex[zindexer]
     assert_array_equal(nparray[npindexer], actual)
 
+    new_data = data.draw(npst.arrays(shape=st.just(actual.shape), dtype=nparray.dtype))
+    nparray[npindexer] = new_data
+    zarray.oindex[zindexer] = new_data
+    assert_array_equal(nparray, zarray[:])
+
 
 @given(data=st.data())
 def test_vindex(data: st.DataObject) -> None:
@@ -63,6 +67,15 @@ def test_vindex(data: st.DataObject) -> None:
     actual = zarray.vindex[indexer]
     assert_array_equal(nparray[indexer], actual)
 
+    # FIXME!
+    # when the indexer is such that a value gets overwritten multiple times,
+    # I think the output depends on chunking. Honestly I don't know why the 1D indexer works.
+    assume(indexer[0].ndim == 1)
+    new_data = data.draw(npst.arrays(shape=st.just(actual.shape), dtype=nparray.dtype))
+    nparray[indexer] = new_data
+    zarray.vindex[indexer] = new_data
+    assert_array_equal(nparray, zarray[:])
+
 
 # @st.composite
 # def advanced_indices(draw, *, shape):