def METHOD_NAME(array_input):
    array = CumlArray(data=array_input)
    assert cp.array_equal(
        cp.asarray(array_input), array.to_output("cupy"), equal_nan=True
    )
    assert np.array_equal(
        cp.asnumpy(array_input), array.to_output("numpy"), equal_nan=True
    )