Python TrainTaskManager.cache_train_jobs Examples

Example #1

def cache_train_jobs():
    """ Cache train jobs."""
    data = request.get_json(silent=True)
    if data is None:
        raise ParamMissError('train_ids')

    train_ids = data.get('train_ids')
    if train_ids is None:
        raise ParamMissError('train_ids')

    processor = TrainTaskManager(DATA_MANAGER)
    cache_result = processor.cache_train_jobs(train_ids)

    return jsonify({'cache_result': cache_result})

Example #2

    def test_cache_train_jobs(self):
        """Test caching train jobs with train ids."""
        train_task_manager = TrainTaskManager(self._mock_data_manager)

        cache_result = train_task_manager.cache_train_jobs(self._train_id_list)
        assert len(self._train_id_list) == len(cache_result)