def test_dataset_setter():
"""Assert that the dataset setter changes the whole dataset."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.dataset = merge(X_class, y_class)
assert atom.dataset.shape == (len(X_class), X_class.shape[1] + 1)
def test_setattr_to_branch():
"""Assert that branch properties can be set from the trainer."""
atom = ATOMClassifier(X_bin, y_bin, random_state=1)
atom.dataset = bin_train
assert atom.shape == (398, 31)