log_vars_test.py 文件源码

def test_log_variables_selected(self):
        """
        Test logging of selected variables from `epoch_data` streams.
        """
        with LogCapture() as log_capture:
            LogVariables(['accuracy', 'loss2']).after_epoch(_EPOCH_ID, _get_epoch_data())

        log_capture.check(
            ('root', 'INFO', '\ttrain accuracy: 1.000000'),
            ('root', 'INFO', '\ttrain loss2:'),
            ('root', 'INFO', '\t\tmean: 1.000000'),
            ('root', 'INFO', '\t\tmedian: 11.000000'),
            ('root', 'INFO', '\ttest accuracy: 2.000000'),
            ('root', 'INFO', '\ttest loss2:'),
            ('root', 'INFO', '\t\tmean: 2.000000'),
            ('root', 'INFO', '\t\tmedian: 22.000000')
        )