| 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) | |