code: def test_confusion_matrix(client, chunks):
      y_true = da.from_array(
          cp.array([2, 0, 2, 2, 0, 1]), chunks=chunks
      )
      y_pred = da.from_array(
          cp.array([0, 0, 2, 2, 0, 2]), chunks=chunks
      )
      cm = confusion_matrix(y_true, y_pred)
      ref = cp.array([[2, 0, 0], [0, 0, 1], [1, 0, 2]])
      cp.testing.assert_array_equal(cm, ref)
  
file_path: cuml/python/cuml/tests/dask/test_dask_metrics.py
imports: import dask.array
  from cuml.dask.metrics import confusion_matrix
  from cuml.testing.utils import stress_param, generate_random_labels
  from sklearn.metrics import confusion_matrix
  import pytest
  from cuml.internals.safe_imports import gpu_only_import
  from itertools import chain, permutations
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 1 for calling a global function with example usage: test_confusion_matrix(client, chunks) without return types