def METHOD_NAME(iris_clean):
    # Arrange
    x, y = iris_clean.data, iris_clean.target
    x["cat_column"] = x["sepal length (cm)"].astype("category")
    x["cat_column"][::2] = np.nan
    x_train, x_test, y_train, y_test = train_test_split(
        x, y,
        test_size=0.3,
        random_state=55
    )
    train_dataset = Dataset(
        pd.concat([x_train, y_train], axis=1),
        features=iris_clean.feature_names + ["cat_column"],
        label='target'
    )
    test_dataset = Dataset(
        pd.concat([x_test, y_test], axis=1),
        features=iris_clean.feature_names + ["cat_column"],
        label='target'
    )
    # Run
    TrainTestSamplesMix().run(
        test_dataset=test_dataset,
        train_dataset=train_dataset
    )