first draft of the variable creation tests

Author: keewis

xarray_array_testing/creation.py

@@ -0,0 +1,17 @@
+import hypothesis.strategies as st
+import xarray.testing.strategies as xrst
+from hypothesis import given
+
+
+class CreationTests:
+    array_type: type
+
+    @staticmethod
+    def array_strategy_fn(*, shape, dtype):
+        raise NotImplementedError
+
+    @given(st.data())
+    def test_create_variable(self, data):
+        variable = data.draw(xrst.variables(array_strategy_fn=self.array_strategy_fn))
+
+        assert isinstance(variable.data, self.array_type)

xarray_array_testing/tests/test_numpy.py

@@ -0,0 +1,17 @@
+import hypothesis.strategies as st
+import numpy as np
+
+from xarray_array_testing.creation import CreationTests
+
+
+def create_numpy_array(*, shape, dtype):
+    return st.builds(np.ones, shape=st.just(shape), dtype=st.just(dtype))
+
+
+class TestCreationNumpy(CreationTests):
+    array_type = np.ndarray
+    array_module = np
+
+    @staticmethod
+    def array_strategy_fn(*, shape, dtype):
+        return create_numpy_array(shape=shape, dtype=dtype)