Test round-tripping of binary reel headers to and from a byte stream.

This commit is contained in:
Robert Smallshire
2015-05-12 10:01:42 +02:00
parent 5bfba47daf
commit d0a1eb5a47
3 changed files with 40 additions and 1 deletions
+24
View File
@@ -0,0 +1,24 @@
import unittest
from hypothesis import given
from io import BytesIO
from hypothesis.specifiers import sampled_from
from segpy.binary_reel_header import BinaryReelHeader
from segpy.header import are_equal
from segpy.toolkit import write_binary_reel_header, read_binary_reel_header
from test.strategies import header
class TestBinaryReelHeader(unittest.TestCase):
@given(header(BinaryReelHeader),
sampled_from(['<', '>']))
def test_roundtrip(self, write_header, endian):
with BytesIO() as write_stream:
write_binary_reel_header(write_stream, write_header, endian)
written_steam = write_stream.getvalue()
with BytesIO(written_steam) as read_stream:
read_header = read_binary_reel_header(read_stream, endian)
self.assertTrue(are_equal(write_header, read_header))