def test_X_train_setter():
"""Assert that the X_train setter changes the training feature set."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
new_X_train = atom.X_train
new_X_train.iloc[0, 0] = 999
atom.X_train = new_X_train
assert atom.X_train.iloc[0, 0] == 999
def test_setter_error_unequal_rows():
"""Assert that an error is raised when the setter has unequal rows."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
with pytest.raises(ValueError, match=r"number of rows"):
atom.X_train = X_bin