Add tests for new core object methods.

Author: pp-mo

tests/unit/core/test_NcData.py

@@ -58,5 +58,61 @@ def test(self, mocker):
         mocker.patch("ncdata.utils.ncdata_copy", mock_copycall)
         ncdata = NcData()
         result = ncdata.copy()
-        assert mock_copycall.called_once_witk(mocker.call(ncdata))
+        assert mock_copycall.call_args_list == [mocker.call(ncdata)]
         assert result == mock_copied_ncdata
+
+
+class Test_NcData_eq:
+    # check that == calls dataset_differences
+    def test(self, mocker):
+        ncdata1, ncdata2 = [NcData(name) for name in ("data1", "data2")]
+        called = mocker.patch("ncdata.utils.dataset_differences")
+        ncdata1 == ncdata2
+        assert called.call_args_list == [mocker.call(ncdata1, ncdata2)]
+
+    def test_self_equal(self, mocker):
+        ds = NcData()
+        called = mocker.patch("ncdata.utils.dataset_differences")
+        assert ds == ds
+        assert called.call_args_list == []
+
+    def test_badtype_nonequal(self, mocker):
+        ds = NcData()
+        called = mocker.patch("ncdata.utils.dataset_differences")
+        assert ds != 1
+        assert called.call_args_list == []
+
+
+class Test_NcVariable_slicer:
+    # check that .slicer makes a slice.
+    def test(self, mocker):
+        ncdata1 = NcData()
+
+        dim_args = (1, 2, 3)  # N.B. not actually acceptable for a real usage
+        mock_return = mocker.sentinel.retval
+        called = mocker.patch("ncdata.utils.Slicer", return_value=mock_return)
+        result = ncdata1.slicer(*dim_args)
+
+        assert called.call_args_list == [mocker.call(ncdata1, *dim_args)]
+        assert result == mock_return
+
+
+class Test_NcVariable_getitem:
+    # check that data[*keys] calls data.slicer[*keys]
+    def test(self, mocker):
+        from ncdata.utils import Slicer
+
+        ncdata1 = NcData()
+
+        dim_keys = (1, 2, 3)  # N.B. not actually acceptable for a real usage
+        mock_slicer = mocker.MagicMock(spec=Slicer)
+        called = mocker.patch(
+            "ncdata._core.NcData.slicer", return_value=mock_slicer
+        )
+        result = ncdata1[*dim_keys]
+
+        assert called.call_args_list == [mocker.call()]
+        assert mock_slicer.__getitem__.call_args_list == [
+            mocker.call(dim_keys)
+        ]
+        assert result == mock_slicer[dim_keys]

tests/unit/core/test_NcVariable.py

@@ -7,6 +7,7 @@
 import dask.array as da
 import numpy as np
 import pytest
+
 from ncdata import NcVariable
 from ncdata.utils import variable_differences
 
@@ -257,3 +258,24 @@ def test_populated(self):
         )
         result = var.copy()
         self.check_var_iscopy(result, var)
+
+
+class Test_NcVariable_eq:
+    def test_equal(self, mocker):
+        # Check that == calls variable_differences
+        var1, var2 = [NcVariable(name=name) for name in ("x", "y")]
+        called = mocker.patch("ncdata.utils.variable_differences")
+        var1 == var2
+        assert called.call_args_list == [mocker.call(var1, var2)]
+
+    def test_self_equal(self, mocker):
+        var = NcVariable(name="x")
+        called = mocker.patch("ncdata.utils.variable_differences")
+        assert var == var
+        assert called.call_args_list == []
+
+    def test_badtype_nonequal(self, mocker):
+        var = NcVariable(name="x")
+        called = mocker.patch("ncdata.utils.variable_differences")
+        assert var != 1
+        assert called.call_args_list == []