test_RegridOperator.py

import datetime
import faulthandler
from importlib.util import find_spec
import unittest

faulthandler.enable()  # to debug seg faults and timeouts

import cf


# ESMF renamed its Python module to `esmpy` at ESMF version 8.4.0. Allow
# either for now for backwards compatibility.
esmpy_imported = False
# Note: here only need esmpy for cf under-the-hood code, not in test
# directly, so no need to actually import esmpy, just test it is there.
if find_spec("esmpy") or find_spec("ESMF"):
    esmpy_imported = True


class RegridOperatorTest(unittest.TestCase):

    def setUp(self):
        src = cf.example_field(0)
        dst = cf.example_field(1)
        self.r = src.regrids(dst, "linear", return_operator=True)

    @unittest.skipUnless(esmpy_imported, "Requires esmpy/ESMF package.")
    def test_RegridOperator_attributes(self):
        self.assertEqual(self.r.coord_sys, "spherical")
        self.assertEqual(self.r.method, "linear")
        self.assertEqual(self.r.dimensionality, 2)
        self.assertEqual(self.r.start_index, 1)
        self.assertTrue(self.r.src_cyclic)
        self.assertFalse(self.r.dst_cyclic)
        self.assertEqual(len(self.r.src_shape), 2)
        self.assertEqual(len(self.r.dst_shape), 2)
        self.assertEqual(len(self.r.src_coords), 2)
        self.assertEqual(len(self.r.src_bounds), 0)
        self.assertIsNone(self.r.src_axes)
        self.assertIsNone(self.r.dst_axes)
        self.assertIsNone(self.r.src_mask)
        self.assertIsNone(self.r.dst_mask)
        self.assertEqual(self.r.weights.ndim, 2)
        self.assertIsNone(self.r.row)
        self.assertIsNone(self.r.col)
        self.assertIsNone(self.r.weights_file)
        self.assertFalse(self.r.src_mesh_location)
        self.assertFalse(self.r.dst_mesh_location)
        self.assertFalse(self.r.src_featureType)
        self.assertFalse(self.r.dst_featureType)
        self.assertIsNone(self.r.src_z)
        self.assertIsNone(self.r.dst_z)
        self.assertFalse(self.r.ln_z)

    @unittest.skipUnless(esmpy_imported, "Requires esmpy/ESMF package.")
    def test_RegridOperator_copy(self):
        self.assertIsInstance(self.r.copy(), self.r.__class__)


if __name__ == "__main__":
    print("Run date:", datetime.datetime.now())
    cf.environment()
    print()
    unittest.main(verbosity=2)