def METHOD_NAME(nrows, ncols, n_neighbors, n_clusters, datatype):
    # Using make_blobs here to check averages and neighborhoods
    X, y = make_blobs(
        n_samples=nrows,
        centers=n_clusters,
        cluster_std=0.01,
        n_features=ncols,
        random_state=0,
    )
    X = X.astype(np.float32)
    y = y.astype(np.float32)
    if datatype == "dataframe":
        X = cudf.DataFrame(X)
        y = cudf.DataFrame(y.reshape(nrows, 1))
    knn_cu = cuKNN(n_neighbors=n_neighbors)
    knn_cu.fit(X, y)
    assert knn_cu.score(X, y) >= 0.9999