def get_initialization_data(self) -> Iterable[InitializationData]:
"""
Return the number of empty assignments specified in --num-assignments
"""
return [
MockTaskRunner.get_mock_assignment_data()
for i in range(self.args.blueprint.num_assignments)
]
def get_mock_assignment_data_array() -> List[InitializationData]:
return [MockTaskRunner.get_mock_assignment_data()]
def get_mock_assignment_data_generator() -> Iterable[InitializationData]:
for _ in range(NUM_GENERATED_ASSIGNMENTS):
yield MockTaskRunner.get_mock_assignment_data()
time.sleep(WAIT_TIME_TILL_NEXT_ASSIGNMENT)
def get_mock_assignment_data_array(self) -> List[InitializationData]:
mock_data = MockTaskRunner.get_mock_assignment_data()
return [mock_data, mock_data]