def test_predict_does_not_change_input(train_data, test_data):
model = EarsC1()
original_train_data = train_data.copy()
original_test_data = test_data.copy()
_ = model.fit(train_data).predict(test_data)
assert_frame_equal(original_train_data, train_data)
assert_frame_equal(original_test_data, test_data)
def test_output_format(train_data, test_data):
model = EarsC1()
original_train_data = train_data.copy()
original_test_data = test_data.copy()
prediction = model.fit(train_data).predict(test_data)
assert set(test_data.columns) == (set(prediction.columns) -
{"alarm", "upperbound"})
def test_validate_data_on_fit(train_data):
model = EarsC1()
with pytest.raises(ValueError):
model.fit(train_data.rename(columns={"n_cases": "wrong_column_name"}))