def test_skip_length():
data = NumpyIO(bytearray(2**21))
for num in [1, 63, 64, 64 * 127, 64 * 128, 63 * 128**2, 64 * 128**2]:
block, _ = writer.make_definitions(np.zeros(num), True)
data.seek(0, 0)
core.skip_definition_bytes(data, num)
assert len(block) == data.tell()
def test_skip_length():
class MockIO:
loc = 0
for num in [1, 63, 64, 64*127, 64*128, 63*128**2, 64*128**2]:
block, _ = writer.make_definitions(np.zeros(num), True)
MockIO.loc = 0
core.skip_definition_bytes(MockIO, num)
assert len(block) == MockIO.loc
def test_skip_length():
class MockIO:
loc = 0
for num in [1, 63, 64, 64*127, 64*128, 63*128**2, 64*128**2]:
block, _ = writer.make_definitions(np.zeros(num), True)
MockIO.loc = 0
core.skip_definition_bytes(MockIO, num)
assert len(block) == MockIO.loc