def test_read_chunks(self): with XZ.open('../data/blob.more.xz') as xz: chunk = xz.read(8) assert chunk == 'Some dat' chunk = xz.read(8) assert chunk == 'a so tha' chunk = xz.read(8) assert chunk == 't we can' chunk = xz.read(8) assert chunk == ' read it' chunk = xz.read(8) assert chunk == ' as mult' chunk = xz.read(8) assert chunk == 'iple chu' chunk = xz.read(8) assert chunk == 'nks'
def setup(self): self.xz = XZ.open('../data/blob.xz')
def test_uncompressed_size(self): assert XZ.uncompressed_size('../data/blob.xz') == 4
def test_read_raise(self, mock_xz): mock_xz.flush = 'data-which-should-never-be-there' with XZ.open('../data/blob.more.xz') as xz: chunk = xz.read(8)
def test_uncompressed_size_raise(self, mock_popen): mock_xz = mock.Mock() mock_xz.communicate = mock.Mock(return_value=['data', 'error']) mock_popen.returncode = 1 mock_popen.return_value = mock_xz XZ.uncompressed_size('../data/blob.xz')
def __upload_byte_size(self, image, image_type): if image_type.is_xz(): return XZ.uncompressed_size(image) return os.path.getsize(image)
def __open_upload_stream(self, image, image_type): if image_type.is_xz(): return XZ.open(image) return open(image)