WIP

Author: dcherian

xarray/testing/strategies.py

@@ -197,6 +197,22 @@ def attrs() -> st.SearchStrategy[Mapping[Hashable, Any]]:
     )
 
 
+def serializable_attrs() -> st.SearchStrategy[Mapping[Hashable, Any]]:
+    """
+    Generates arbitrary valid attributes dictionaries for xarray objects.
+
+    These are intended to be serialized, and so, are less general than the
+    `attrs` function above.
+
+    Requires the hypothesis package to be installed.
+
+    See Also
+    --------
+    :ref:`testing.hypothesis`_
+    """
+    return st.dictionaries(_attr_keys, _attr_values)
+
+
 @st.composite
 def variables(
     draw: st.DrawFn,

xarray/tests/test_state_machine.py

@@ -42,6 +42,7 @@ def unique(draw, strategy):
 DIM_NAME = xrst.dimension_names(name_strategy=UNIQUE_NAME, min_dims=1, max_dims=1)
 
 
+# TODO: add datetime64[ns]
 def pandas_index_dtypes() -> st.SearchStrategy[np.dtype]:
     return (
         npst.integer_dtypes(endianness="<", sizes=(32, 64))
@@ -83,6 +84,7 @@ def stack(self, newname):
 
     @rule()
     def unstack(self):
+        # TODO: Drop duplicates
         self.dataset = self.dataset.unstack()
 
     @rule(newname=UNIQUE_NAME)
@@ -128,5 +130,5 @@ def assert_invariants(self):
         _assert_internal_invariants(self.dataset, self.check_default_indexes)
 
 
-DatasetStateMachine.TestCase.settings = settings(max_examples=1000)
+DatasetStateMachine.TestCase.settings = settings(max_examples=1000, deadline=None)
 DatasetTest = DatasetStateMachine.TestCase