def test_inline_summary_without_validation(self): with catch_stdout() as out: network = algorithms.GradientDescent((2, 3, 1), verbose=True) summary = InlineSummary(network) network.last_epoch = 12 network.training.epoch_time = 0.1 network.errors.append(10) network.validation_errors.append(None) summary.show_last() terminal_output = out.getvalue() # training error is 10 self.assertIn("10", terminal_output) # 0.1 sec self.assertIn("0.1", terminal_output) # 12th epoch self.assertIn("12", terminal_output) # No reference to validation error in the last line output_lines = terminal_output.split('\n') last_output_line = output_lines[-2] self.assertNotIn("None", last_output_line)
def test_inline_summary_with_validation(self): with catch_stdout() as out: network = algorithms.GradientDescent((2, 3, 1), verbose=True) summary = InlineSummary(network) network.last_epoch = 12 network.training.epoch_time = 0.1 network.errors.append(10) network.validation_errors.append(20) summary.show_last() summary.finish() terminal_output = out.getvalue() # training error is 10 self.assertIn("10", terminal_output) # validation error is 20 self.assertIn("20", terminal_output) # 0.1 sec self.assertIn("0.1", terminal_output) # 12th epoch self.assertIn("12", terminal_output)