mirror of
https://github.com/wassname/segpy.git
synced 2026-09-10 12:36:21 +08:00
Test round-tripping of binary reel headers to and from a byte stream.
This commit is contained in:
@@ -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))
|
||||
Reference in New Issue
Block a user