def METHOD_NAME(spark, array):
    log = spark.createDataFrame(data=[[1], [2], [3]], schema=["test"])
    assert sorted(
        list(utils.get_unique_entities(array or log, "test").toPandas()["test"])
    ) == [1, 2, 3]