def test_function_not_found(self):
with self.assertRaises(ValueError) as cm:
load_custom_func("tests.test_custom:my_custom_fun")
self.assertEqual(
'function "tests.test_custom:my_custom_fun" not found,'
' unknown attribute "my_custom_fun"', f"{cm.exception}")
def test_not_a_function(self):
with self.assertRaises(ValueError) as cm:
load_custom_func("tests.test_custom:my_custom_var")
self.assertEqual('"tests.test_custom:my_custom_var" is not callable',
f"{cm.exception}")
def test_module_not_found(self):
with self.assertRaises(ValueError) as cm:
load_custom_func("test.test_custom:my_custom_func")
self.assertEqual(
'module for function "test.test_custom:my_custom_func" not found',
f"{cm.exception}")
def test_invalid_id(self):
with self.assertRaises(ValueError) as cm:
load_custom_func("tests.test_custom:my-custom-func")
self.assertEqual(
'func_ref "tests.test_custom:my-custom-func" is invalid,'
' "my-custom-func" is not a valid identifier', f"{cm.exception}")
def test_invalid_ref(self):
with self.assertRaises(ValueError) as cm:
load_custom_func("tests.test_custom.my_custom_func")
self.assertEqual(
'func_ref "tests.test_custom.my_custom_func" is invalid,'
' format must be <module>:<function>', f"{cm.exception}")
def test_ok_class_method(self):
func = load_custom_func(
"tests.test_custom:LoadCustomFunctionTest.my_custom_func")
self.assertIsInstance(func(xr.Dataset()), xr.Dataset)
def test_ok(self):
func = load_custom_func("tests.test_custom:my_custom_func")
self.assertIsInstance(func(xr.Dataset()), xr.Dataset)