experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add summary test verifying that the new summary ops play nicely with variables.

Browse Source 
No core code change.
Change: 136782838
This commit is contained in:
Dan Mané 2016-10-20 16:41:02 -08:00 committed by TensorFlower Gardener
parent dd845147f1
commit 2da2ae76cf
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/python/summary/summary_test.py
View File

@ -40,6 +40,21 @@ class ScalarSummaryTest(tf.test.TestCase):
self.assertEqual(values[0].tag, 'outer/inner')
self.assertEqual(values[0].simple_value, 3.0)
def testSummarizingVariable(self):
with self.test_session() as s:
c = tf.constant(42.0)
v = tf.Variable(c)
ss = tf.summary.scalar('summary', v)
init = tf.initialize_all_variables()
s.run(init)
summ_str = s.run(ss)
summary = tf.Summary()
summary.ParseFromString(summ_str)
self.assertEqual(len(summary.value), 1)
value = summary.value[0]
self.assertEqual(value.tag, 'summary')
self.assertEqual(value.simple_value, 42.0)
def testImageSummary(self):
with self.test_session() as s:
i = tf.ones((5, 4, 4, 3))