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_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
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
