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_write_record(self): file_handle = io.BytesIO() _TFRecordUtil.write_record(file_handle, b'foo') self.assertEqual(self.record, file_handle.getvalue())
def test_write_record(self): file_handle = cStringIO.StringIO() _TFRecordUtil.write_record(file_handle, 'foo') self.assertEqual(self.record, file_handle.getvalue())
def test_write_record(self): file_handle = cStringIO.StringIO() _TFRecordUtil.write_record(file_handle, 'foo') self.assertEqual(self.record, file_handle.getvalue())
def test_write_record(self): file_handle = io.BytesIO() _TFRecordUtil.write_record(file_handle, b'foo') self.assertEqual(self.record, file_handle.getvalue())