def test_logger_collection():
mock1 = MagicMock()
mock2 = MagicMock()
logger = LoggerCollection([mock1, mock2])
assert logger[0] == mock1
assert logger[1] == mock2
assert logger.experiment[0] == mock1.experiment
assert logger.experiment[1] == mock2.experiment
assert logger.save_dir is None
logger.update_agg_funcs({'test': np.mean}, np.sum)
mock1.update_agg_funcs.assert_called_once_with({'test': np.mean}, np.sum)
mock2.update_agg_funcs.assert_called_once_with({'test': np.mean}, np.sum)
logger.agg_and_log_metrics({'test': 2.0}, 4)
mock1.agg_and_log_metrics.assert_called_once_with({'test': 2.0}, 4)
mock2.agg_and_log_metrics.assert_called_once_with({'test': 2.0}, 4)
logger.close()
mock1.close.assert_called_once()
mock2.close.assert_called_once()
def test_logger_collection():
mock1 = MagicMock()
mock2 = MagicMock()
logger = LoggerCollection([mock1, mock2])
assert logger[0] == mock1
assert logger[1] == mock2
assert logger.experiment[0] == mock1.experiment
assert logger.experiment[1] == mock2.experiment
logger.close()
mock1.close.assert_called_once()
mock2.close.assert_called_once()
