def test_python_pack_used_as_fallback(self, python_packer, mgr):
# Don't let the manager report that it contains 'cpp'
mgr.__contains__ = MagicMock(return_value=False)
data = byte_arrays_of_floats()
with test.util.force_python_ibm_float(False):
ibm_float_packer.pack_ibm_floats(data)
python_packer.return_value.pack.assert_called_with(data)
def test_python_pack_used_as_fallback(self, python_packer, mgr):
# Don't let the manager report that it contains 'cpp'
mgr.__contains__ = MagicMock(return_value=False)
data = byte_arrays_of_floats()
with test.util.force_python_ibm_float(False):
ibm_float_packer.pack_ibm_floats(data)
python_packer.return_value.pack.assert_called_with(data)
def test_python_pack_used_when_forced(self, python_packer, mgr):
# Let the manager report that it contains 'cpp'
mgr.__contains__ = MagicMock(return_value=True)
data = byte_arrays_of_floats().example()
with test.util.force_python_ibm_float(True):
ibm_float_packer.pack_ibm_floats(data)
mgr.__getitem__.assert_not_called()
python_packer.return_value.pack.assert_called_with(data)
def test_python_pack_used_when_forced(self, python_packer, mgr):
# Let the manager report that it contains 'cpp'
mgr.__contains__ = MagicMock(return_value=True)
data = byte_arrays_of_floats().example()
with test.util.force_python_ibm_float(True):
ibm_float_packer.pack_ibm_floats(data)
mgr.__getitem__.assert_not_called()
python_packer.return_value.pack.assert_called_with(data)
def test_roundtrip(self, data):
packed = ibm_float_packer.pack_ibm_floats(data)
unpacked = ibm_float_packer.unpack_ibm_floats(packed, len(data))
for f, u in zip(data, unpacked):
assert almost_equal(
f, float(u),
epsilon=EPSILON_IBM_FLOAT)
def write_binary_values(fh, values, seg_y_type, endian='>'):
"""Write a series of values to a file.
Args:
fh: A file-like-object open for writing in binary mode.
values: An iterable series of values.
seg_y_type: The SEG Y data type.
endian: '>' for big-endian data (the standard and default),
'<' for little-endian (non-standard)
"""
ctype = SEG_Y_TYPE_TO_CTYPE[seg_y_type]
buf = (pack_ibm_floats(values)
if ctype == 'ibm'
else pack_values(values, ctype, endian))
fh.write(buf)
def test_roundtrip(self, floats):
byte_data, num_items = floats
unpacked = ibm_float_packer.unpack_ibm_floats(byte_data, num_items)
packed = ibm_float_packer.pack_ibm_floats(unpacked)
assert bytes(byte_data) == bytes(packed)
def test_unpack_empty(self):
dest = ibm_float_packer.pack_ibm_floats(b'')
assert len(dest) == 0
def test_roundtrip(self, data):
packed = ibm_float_packer.pack_ibm_floats(data)
unpacked = ibm_float_packer.unpack_ibm_floats(packed, len(data))
for f, u in zip(data, unpacked):
assert almost_equal(f, float(u), epsilon=EPSILON_IBM_FLOAT)
def test_packed_floats_are_four_bytes(self, floats):
packed = ibm_float_packer.pack_ibm_floats(floats)
assert len(packed) == len(floats) * 4
def test_roundtrip(self, floats):
byte_data, num_items = floats
unpacked = ibm_float_packer.unpack_ibm_floats(byte_data, num_items)
packed = ibm_float_packer.pack_ibm_floats(unpacked)
assert bytes(byte_data) == bytes(packed)
def test_unpack_empty(self):
dest = ibm_float_packer.pack_ibm_floats(b'')
assert len(dest) == 0
def test_packed_floats_are_four_bytes(self, floats):
packed = ibm_float_packer.pack_ibm_floats(floats)
assert len(packed) == len(floats) * 4