From a14aeb51f6787a93ce12958bc3e9634fb0e84cdb Mon Sep 17 00:00:00 2001 From: Justus Magin Date: Sat, 20 May 2023 14:34:18 +0200 Subject: [PATCH] use `full_like` to construct the expected result --- xarray/tests/test_ufuncs.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/xarray/tests/test_ufuncs.py b/xarray/tests/test_ufuncs.py index 6b4c3f38ee9..774b729b4fd 100644 --- a/xarray/tests/test_ufuncs.py +++ b/xarray/tests/test_ufuncs.py @@ -4,7 +4,7 @@ import pytest import xarray as xr -from xarray.tests import assert_allclose, assert_array_equal, mock +from xarray.tests import assert_array_equal, mock from xarray.tests import assert_identical as assert_identical_ @@ -25,7 +25,12 @@ def assert_identical(a, b): ], ) def test_unary(a): - assert_allclose(a + 1, np.cos(a)) + fill_value = np.cos(0) + + expected = xr.full_like(a, fill_value=fill_value, dtype=fill_value.dtype) + actual = np.cos(a) + + assert_identical(actual, expected) def test_binary():