def test_get_current_task_returns_last_time_units_element(): processor = Processor() processor.add_time_unit() t = Task('T1', 1, 1) processor.set_task(t) assert processor.get_current_task() == t
def test_get_current_task_returns_none_when_time_units_is_empty(): processor = Processor() assert processor.get_current_task() is None