add tests

Author: Joseph Hamman

setup.cfg

@@ -8,7 +8,7 @@ select = B,C,E,F,W,T4,B9
 
 [isort]
 known_first_party=xarray_schema
-known_third_party=numpy,pkg_resources,setuptools,xarray
+known_third_party=numpy,pkg_resources,pytest,setuptools,xarray
 multi_line_output=3
 include_trailing_comma=True
 force_grid_wrap=0

tests/test_core.py

@@ -1,9 +1,94 @@
+import numpy as np
+import pytest
+import xarray as xr
+
 from xarray_schema import DataArraySchema, DatasetSchema
+from xarray_schema.core import SchemaError
 
 
-def test_empyt_constructors():
+def test_dataarray_empty_constructor():
 
     da_schema = DataArraySchema()
     assert hasattr(da_schema, 'validate')
+
+
+def test_dataarray_validate_dtype():
+
+    da = xr.DataArray(np.ones(4, dtype='i4'))
+    schema = DataArraySchema(dtype='i4')
+    schema.validate(da)
+
+    schema = DataArraySchema(dtype=np.int32)
+    schema.validate(da)
+
+    schema = DataArraySchema(dtype=np.integer)
+    schema.validate(da)
+
+    schema = DataArraySchema(dtype=np.floating)
+    with pytest.raises(SchemaError, match=r'.*floating.*'):
+        schema.validate(da)
+
+
+def test_dataarray_validate_name():
+
+    da = xr.DataArray(np.ones(4), name='foo')
+    schema = DataArraySchema(name='foo')
+    schema.validate(da)
+
+    schema = DataArraySchema(name='bar')
+    with pytest.raises(SchemaError, match=r'.*foo.*'):
+        schema.validate(da)
+
+
+def test_dataarray_validate_shape():
+
+    da = xr.DataArray(np.ones(4))
+    schema = DataArraySchema(shape=(4,))
+    schema.validate(da)
+
+    schema = DataArraySchema(shape=(4, 2))
+    with pytest.raises(SchemaError, match=r'.*ndim.*'):
+        schema.validate(da)
+
+    schema = DataArraySchema(shape=(3,))
+    with pytest.raises(SchemaError, match=r'.*(4).*'):
+        schema.validate(da)
+
+
+def test_dataarray_validate_dims():
+
+    da = xr.DataArray(np.ones(4), dims=['x'])
+    schema = DataArraySchema(dims=['x'])
+    schema.validate(da)
+
+    schema = DataArraySchema(dims=(['x', 'y']))
+    with pytest.raises(SchemaError, match=r'.*length of dims.*'):
+        schema.validate(da)
+
+    schema = DataArraySchema(dims=['y'])
+    with pytest.raises(SchemaError, match=r'.*(y).*'):
+        schema.validate(da)
+
+
+def test_dataset_empty_constructor():
     ds_schema = DatasetSchema()
     assert hasattr(ds_schema, 'validate')
+
+
+def test_dataset_example():
+
+    ds = xr.Dataset(
+        {
+            'x': xr.DataArray(np.arange(4) - 2, dims='x'),
+            'foo': xr.DataArray(np.ones(4, dtype='i4'), dims='x'),
+            'bar': xr.DataArray(np.arange(8, dtype=np.float32).reshape(4, 2), dims=('x', 'y')),
+        }
+    )
+
+    ds_schema = DatasetSchema(
+        {
+            'foo': DataArraySchema(name='foo', dtype=np.int32, dims=['x']),
+            'bar': DataArraySchema(name='bar', dtype=np.floating, dims=['x', 'y']),
+        }
+    )
+    ds_schema.validate(ds)