File size: 349 Bytes
8b414b0 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from src.data_reader import load_train_test_df
from src.solutions.constant_predictor import ConstantPredictorSolution
def test_constant_predictor():
_, test_df = load_train_test_df(is_testing=True)
predictor = ConstantPredictorSolution()
submission_df = predictor.predict(test_df)
assert submission_df.shape == (len(test_df), 7)
|