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.finalize("success") mock1.finalize.assert_called_once() mock2.finalize.assert_called_once()
def test_logger_collection(): mock1 = MagicMock() mock2 = MagicMock() with pytest.deprecated_call(match="`LoggerCollection` is deprecated in v1.6"): 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.log_metrics(metrics={"test": 2.0}, step=4) mock1.log_metrics.assert_called_once_with(metrics={"test": 2.0}, step=4) mock2.log_metrics.assert_called_once_with(metrics={"test": 2.0}, step=4) logger.finalize("success") mock1.finalize.assert_called_once() mock2.finalize.assert_called_once()