Fix tests

rapidsai · Mar 18, 2021 · b5054f9 · b5054f9
1 parent 97c819d
commit b5054f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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