def test_multi_sync_env_runner_info_no_data_logger(mock_task, mock_agent): # Assign mock_tasks: List[TaskType] = [mock_task, mock_task] env_runner = MultiSyncEnvRunner(mock_tasks, mock_agent) env_runner.logger = mock.MagicMock() info_data = dict(episodes=[2], iterations=[10], scores=[1], mean_scores=[2], epsilons=[1]) # Act env_runner.info(**info_data) # Assert env_runner.logger.info.assert_called_once()
def test_multi_sync_env_runner_info_with_data_logger(mock_task, mock_agent): # Assign data_logger = mock.MagicMock() env_runner = MultiSyncEnvRunner(mock_task, mock_agent, data_logger=data_logger) env_runner.logger = mock.MagicMock() info_data = dict(episodes=[2], iterations=[10], scores=[1], mean_scores=[2], epsilons=[1]) # Act env_runner.info(**info_data) # Assert env_runner.logger.info.assert_called_once() assert data_logger.log_value.call_count == 4 mock_agent.log_metrics.assert_called_once_with(data_logger, mock.ANY, full_log=False)