diff options
| -rw-r--r-- | calendar_manager.py | 4 | ||||
| -rw-r--r-- | test.py | 8 |
2 files changed, 12 insertions, 0 deletions
diff --git a/calendar_manager.py b/calendar_manager.py index 318eb83..9ef6596 100644 --- a/calendar_manager.py +++ b/calendar_manager.py @@ -132,6 +132,10 @@ class CalendarManager: return self.entries def correct_dates(self, list_of_events): + for entry in self.entries: + entry.corrected_start_date = None + entry.corrected_end_date = None + for start_date, end_date, original_id in list_of_events: entry = self.get_entry_by_id(original_id) entry.corrected_start_date = start_date @@ -36,5 +36,13 @@ def test_calculate_prediction(): print(f"Predicted completion date: {prediction}") print(calendar_manager.list_entries()) + prediction_controller.make_prediction("2023-02-01", 2) + + prediction = prediction_controller.get_prediction() + + # Output the result + print(f"Predicted completion date: {prediction}") + print(calendar_manager.list_entries()) + # Run the test test_calculate_prediction()
\ No newline at end of file |