def test_compatibility_read_write(self):
for record in [b'', b'blah', b'another blah']:
file_handle = io.BytesIO()
_TFRecordUtil.write_record(file_handle, record)
file_handle.seek(0)
actual = _TFRecordUtil.read_record(file_handle)
self.assertEqual(record, actual)
def test_compatibility_read_write(self):
for record in ['', 'blah', 'another blah']:
file_handle = cStringIO.StringIO()
_TFRecordUtil.write_record(file_handle, record)
file_handle.reset()
actual = _TFRecordUtil.read_record(file_handle)
self.assertEqual(record, actual)
def test_compatibility_read_write(self):
for record in ['', 'blah', 'another blah']:
file_handle = cStringIO.StringIO()
_TFRecordUtil.write_record(file_handle, record)
file_handle.reset()
actual = _TFRecordUtil.read_record(file_handle)
self.assertEqual(record, actual)
def test_compatibility_read_write(self):
for record in ['', 'blah', 'another blah']:
file_handle = io.BytesIO()
_TFRecordUtil.write_record(file_handle, record)
file_handle.seek(0)
actual = _TFRecordUtil.read_record(file_handle)
self.assertEqual(record, actual)
def test_read_record(self): actual = _TFRecordUtil.read_record(self._as_file_handle(self.record)) self.assertEqual(b'foo', actual)
def _test_error(self, record, error_text):
with self.assertRaisesRegex(ValueError, re.escape(error_text)):
_TFRecordUtil.read_record(self._as_file_handle(record))
def _test_error(self, record, error_text):
with self.assertRaises(ValueError) as context:
_TFRecordUtil.read_record(self._as_file_handle(record))
self.assertIn(error_text, context.exception.message)
def test_read_record(self):
actual = _TFRecordUtil.read_record(self._as_file_handle(self.record))
self.assertEqual('foo', actual)
def _test_error(self, record, error_text):
with self.assertRaises(ValueError) as context:
_TFRecordUtil.read_record(self._as_file_handle(record))
self.assertIn(error_text, context.exception.message)
def _test_error(self, record, error_text):
with self.assertRaisesRegexp(ValueError, re.escape(error_text)):
_TFRecordUtil.read_record(self._as_file_handle(record))
