Add test

bouweandela · Apr 17, 2024 · 27ecc8f · 27ecc8f
1 parent 0d1fc27
commit 27ecc8f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/iris/tests/unit/fileformats/netcdf/loader/test__get_cf_var_data.py b/lib/iris/tests/unit/fileformats/netcdf/loader/test__get_cf_var_data.py
@@ -10,7 +10,7 @@
 
 from unittest import mock
 
-from dask.array import Array as dask_array
+import dask.array as da
 import numpy as np
 
 from iris._lazy_data import _optimum_chunksize
@@ -50,7 +50,8 @@ def test_cf_data_type(self):
         chunks = [1, 12, 100]
         cf_var = self._make(chunks)
         lazy_data = _get_cf_var_data(cf_var, self.filename)
-        self.assertIsInstance(lazy_data, dask_array)
+        self.assertIsInstance(lazy_data, da.Array)
+        self.assertIsInstance(da.utils.meta_from_array(lazy_data), np.ma.MaskedArray)
 
     def test_cf_data_chunks(self):
         chunks = [2500, 240, 200]
@@ -90,7 +91,7 @@ def test_cf_data_contiguous(self):
     def test_type__1kf8_is_lazy(self):
         cf_var = self._make(shape=(1000,), dtype="f8")
         var_data = _get_cf_var_data(cf_var, self.filename)
-        self.assertIsInstance(var_data, dask_array)
+        self.assertIsInstance(var_data, da.Array)
 
     def test_arraytype__1ki2_is_real(self):
         cf_var = self._make(shape=(1000,), dtype="i2")