def test_print_skipped_if_not_verbose(self, capsys): from skorch.callbacks import PrintLog print_log = PrintLog().initialize() net = Mock(history=[{'loss': 123}], verbose=0) print_log.on_epoch_end(net) stdout = capsys.readouterr()[0] assert not stdout
def test_print_not_skipped_if_verbose(self, capsys): from skorch.callbacks import PrintLog print_log = PrintLog().initialize() net = Mock(history=[{'loss': 123}], verbose=1) print_log.on_epoch_end(net) stdout = capsys.readouterr()[0] result = [x.strip() for x in stdout.split()] expected = ['loss', '------', '123'] assert result == expected