def test_set_fields(self):
msg = IterationMessage()
# setting iter will clear tic and iter output fields
msg.current_iter = 3
self.assertGreater(msg._current_iter_tic, 0.0)
self.assertEqual(msg._current_iter_output, None)
# setting iter output will update iter duration
msg.current_iter_output = {CONSOLE: {'test': 'test'}}
self.assertEqual(msg.current_iter, 3)
self.assertGreater(msg.iter_duration, 0.0)
self.assertRegexpMatches(msg.to_console_string(), '.*test=test.*')
with self.assertRaisesRegexp(ValueError, ''):
msg.current_iter = 'test'
def test_set_fields(self):
msg = IterationMessage()
# setting iter will clear tic and iter output fields
msg.current_iter = 3
self.assertGreater(msg._current_iter_tic, 0.0)
self.assertEqual(msg._current_iter_output, None)
# setting iter output will update iter duration
msg.current_iter_output = {CONSOLE: {'test': 'test'}}
self.assertEqual(msg.current_iter, 3)
self.assertGreater(msg.iter_duration, 0.0)
self.assertRegexpMatches(msg.to_console_string(), '.*test=test.*')
with self.assertRaisesRegexp(ValueError, ''):
msg.current_iter = 'test'
def test_interfaces(self):
msg = IterationMessage()
msg.current_iter = 0
self.assertEqual(msg.current_iter, 0)
self.assertEqual(msg.ops_to_run, {})
self.assertEqual(msg.data_feed_dict, {})
self.assertEqual(msg.current_iter_output, None)
self.assertEqual(msg.should_stop, None)
self.assertEqual(msg.phase, TRAIN)
self.assertEqual(msg.is_training, True)
self.assertEqual(msg.is_validation, False)
self.assertEqual(msg.is_inference, False)
msg.current_iter_output = {'test'}
self.assertEqual(msg.current_iter_output, {'test'})
self.assertGreater(msg.iter_duration, 0.0)
self.assertStartsWith(msg.to_console_string(), 'training')
self.assertEqual(msg.to_tf_summary(0), None)
def test_interfaces(self):
msg = IterationMessage()
msg.current_iter = 0
self.assertEqual(msg.current_iter, 0)
self.assertEqual(msg.ops_to_run, {})
self.assertEqual(msg.data_feed_dict, {})
self.assertEqual(msg.current_iter_output, None)
self.assertEqual(msg.should_stop, False)
self.assertEqual(msg.phase, TRAIN)
self.assertEqual(msg.is_training, True)
self.assertEqual(msg.is_validation, False)
self.assertEqual(msg.is_inference, False)
msg.current_iter_output = {'test'}
self.assertEqual(msg.current_iter_output, {'test'})
self.assertGreater(msg.iter_duration, 0.0)
self.assertStartsWith(msg.to_console_string(), 'Training')
self.assertEqual(msg.to_tf_summary(0), None)