Update cuDF's assert_eq import (#239)

rapidsai/cudf#16063 has updated the import location of `assert_eq` to the public `cudf.testing.assert_eq`, this change updates imports accordingly.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Mads R. B. Kristensen (https://github.com/madsbk)

URL: #239

python/distributed-ucxx/distributed_ucxx/tests/test_ucxx.py

@@ -185,7 +185,7 @@ async def test_ping_pong_cudf(ucxx_loop, g):
     # *** ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `CXXABI_1.3.11'
     # not found (required by python3.7/site-packages/pyarrow/../../../libarrow.so.12)
     cudf = pytest.importorskip("cudf")
-    from cudf.testing._utils import assert_eq
+    from cudf.testing import assert_eq
 
     cudf_obj = g(cudf)
 

python/ucxx/_lib_async/tests/test_custom_send_recv.py

@@ -118,7 +118,7 @@ async def serve_forever(ep):
     typ = type(msg)
     res = typ.deserialize(ucx_header, cudf_buffer)
 
-    from cudf.testing._utils import assert_eq
+    from cudf.testing import assert_eq
 
     assert_eq(res, msg)
     await uu.comm.ep.close()

python/ucxx/_lib_async/tests/test_send_recv_two_workers.py

@@ -94,7 +94,7 @@ async def read():
     if isinstance(rx_cuda_obj, cupy.ndarray):
         cupy.testing.assert_allclose(rx_cuda_obj, pure_cuda_obj)
     else:
-        from cudf.testing._utils import assert_eq
+        from cudf.testing import assert_eq
 
         assert_eq(rx_cuda_obj, pure_cuda_obj)