From f41959ccb2d9d4c722fe8fc3351401d53bcf4900 Mon Sep 17 00:00:00 2001 From: Manjunath Kudlur Date: Fri, 6 Nov 2015 16:27:58 -0800 Subject: TensorFlow: Initial commit of TensorFlow library. TensorFlow is an open source software library for numerical computation using data flow graphs. Base CL: 107276108 --- tensorflow/python/client/events_writer_test.py | 54 ++++++++++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100644 tensorflow/python/client/events_writer_test.py (limited to 'tensorflow/python/client/events_writer_test.py') diff --git a/tensorflow/python/client/events_writer_test.py b/tensorflow/python/client/events_writer_test.py new file mode 100644 index 0000000000..60bce49b1f --- /dev/null +++ b/tensorflow/python/client/events_writer_test.py @@ -0,0 +1,54 @@ +"""Tests for the SWIG-wrapped events writer.""" +import os.path + +from tensorflow.core.framework import summary_pb2 +from tensorflow.core.util import event_pb2 +from tensorflow.python import pywrap_tensorflow +from tensorflow.python.lib.io import tf_record +from tensorflow.python.framework import test_util +from tensorflow.python.platform import googletest + + +class PywrapeventsWriterTest(test_util.TensorFlowTestCase): + + def testWriteEvents(self): + file_prefix = os.path.join(self.get_temp_dir(), "events") + writer = pywrap_tensorflow.EventsWriter(file_prefix) + filename = writer.FileName() + event_written = event_pb2.Event( + wall_time=123.45, step=67, + summary=summary_pb2.Summary( + value=[summary_pb2.Summary.Value(tag="foo", simple_value=89.0)])) + writer.WriteEvent(event_written) + writer.Flush() + writer.Close() + + with self.assertRaises(IOError): + for r in tf_record.tf_record_iterator(filename + "DOES_NOT_EXIST"): + self.assertTrue(False) + + reader = tf_record.tf_record_iterator(filename) + event_read = event_pb2.Event() + + event_read.ParseFromString(next(reader)) + self.assertTrue(event_read.HasField("file_version")) + + event_read.ParseFromString(next(reader)) + # Second event + self.assertProtoEquals(""" + wall_time: 123.45 step: 67 + summary { value { tag: 'foo' simple_value: 89.0 } } + """, event_read) + + with self.assertRaises(StopIteration): + next(reader) + + def testWriteEventInvalidType(self): + class _Invalid(object): + def __str__(self): return "Invalid" + with self.assertRaisesRegexp(TypeError, "Invalid"): + pywrap_tensorflow.EventsWriter("foo").WriteEvent(_Invalid()) + + +if __name__ == "__main__": + googletest.main() -- cgit v1.2.3