From b5054f9c85ea6c1b979329db2b198db8e1afcf75 Mon Sep 17 00:00:00 2001 From: Ashwin Srinath Date: Thu, 18 Mar 2021 15:36:50 -0400 Subject: [PATCH] Fix tests --- python/cudf/cudf/tests/test_scalar.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cudf/cudf/tests/test_scalar.py b/python/cudf/cudf/tests/test_scalar.py index 1a8aca2b2a3..a42edadc103 100644 --- a/python/cudf/cudf/tests/test_scalar.py +++ b/python/cudf/cudf/tests/test_scalar.py @@ -294,9 +294,9 @@ def test_device_scalar_direct_construction(value): @pytest.mark.parametrize("value", SCALAR_VALUES) def test_construct_from_scalar(value): value = cudf.utils.utils.to_cudf_compatible_scalar(value) - x = cudf.Scalar(1, value.dtype) + x = cudf.Scalar(value, value.dtype) y = cudf.Scalar(x) - assert x.value == y.value + assert x.value == y.value or np.isnan(x.value) and np.isnan(y.value) # check that this works: y.device_value