def test_data_properties_to_df():
"""Assert that the data attributes are converted to a df at setter."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.X = X_bin_array
assert isinstance(atom.X, pd.DataFrame)
def test_X_setter():
"""Assert that the X setter changes the feature set."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.X = atom.X.iloc[:, :10]
assert atom.X.shape == (len(X_bin), 10)