Add back test_struct_scalar_null

rapidsai · Jan 17, 2025 · 8472cf8 · 8472cf8
1 parent d146229
commit 8472cf8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/python/cudf/cudf/core/scalar.py b/python/cudf/cudf/core/scalar.py
@@ -365,6 +365,7 @@ def value(self):
             self._device_value_to_host()
         return self._host_value
 
+    # TODO: change to @functools.cached_property
     @property
     def dtype(self):
         if self._host_dtype is not None:

diff --git a/python/cudf/cudf/tests/test_struct.py b/python/cudf/cudf/tests/test_struct.py
@@ -194,6 +194,11 @@ def test_struct_scalar_host_construction_no_dtype_inference(data, dtype):
     assert slr.value == data
 
 
+def test_struct_scalar_null():
+    slr = cudf.Scalar(cudf.NA, dtype=cudf.StructDtype)
+    assert cudf.Scalar.from_pylibcudf(slr.device_value).value is cudf.NA
+
+
 def test_struct_explode():
     s = cudf.Series([], dtype=cudf.StructDtype({}))
     expect = cudf.DataFrame({})