diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 12 |
1 files changed, 6 insertions, 6 deletions
@@ -19,13 +19,13 @@ def test_calculate_prediction(): date_calculator = DateCalculator() calendar_manager = CalendarManager() - prediction_controller = PredictionController(calendar_manager, date_calculator, ["Erziehungszeit 100%", "Erziehungszeit 50%", "Erziehungszeit pauschal", "Sonstige"]) + prediction_controller = PredictionController(calendar_manager, date_calculator, ["full_project", "half_project", "event"]) - calendar_manager.add_entry(event1_str[0], event1_str[1], "Sonstige") - calendar_manager.add_entry(event2_str[0], event2_str[1], "Sonstige") - calendar_manager.add_entry(project1_str[0], project1_str[1], "Erziehungszeit 100%") - calendar_manager.add_entry(project2_str[0], project2_str[1], "Erziehungszeit 50%") - calendar_manager.add_entry(project3_str[0], project3_str[1], "Erziehungszeit 50%") + calendar_manager.add_entry(event1_str[0], event1_str[1], "event") + calendar_manager.add_entry(event2_str[0], event2_str[1], "event") + calendar_manager.add_entry(project1_str[0], project1_str[1], "full_project") + calendar_manager.add_entry(project2_str[0], project2_str[1], "half_project") + calendar_manager.add_entry(project3_str[0], project3_str[1], "half_project") # Set launch date and duration prediction_controller.make_prediction("2023-01-01", 2) |