mirror of
https://github.com/google/brotli.git
synced 2026-02-07 13:51:45 +00:00
f81d6bc7f0
Avoid file IO. Drive-by: drop bro.py and bro_test.py; we do not support it well and likely no one uses it. PiperOrigin-RevId: 834206605
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
# Copyright 2016 The Brotli Authors. All rights reserved.
|
|
#
|
|
# Distributed under MIT license.
|
|
# See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
|
|
|
|
import brotli
|
|
import pytest
|
|
|
|
from . import _test_utils
|
|
|
|
|
|
@pytest.mark.parametrize("quality", [1, 6, 9, 11])
|
|
@pytest.mark.parametrize("text_name", _test_utils.gather_text_inputs())
|
|
def test_single_process(quality, text_name):
|
|
original = _test_utils.take_input(text_name)
|
|
compressor = brotli.Compressor(quality=quality)
|
|
compressed = compressor.process(original)
|
|
compressed += compressor.finish()
|
|
decompressed = brotli.decompress(compressed)
|
|
assert original == decompressed
|
|
|
|
|
|
@pytest.mark.parametrize("quality", [1, 6, 9, 11])
|
|
@pytest.mark.parametrize("text_name", _test_utils.gather_text_inputs())
|
|
def test_multiple_process(quality, text_name):
|
|
original = _test_utils.take_input(text_name)
|
|
chunk_size = 2048
|
|
chunks = _test_utils.chunk_input(original, chunk_size)
|
|
compressor = brotli.Compressor(quality=quality)
|
|
compressed = b''
|
|
for chunk in chunks:
|
|
compressed += compressor.process(chunk)
|
|
compressed += compressor.finish()
|
|
decompressed = brotli.decompress(compressed)
|
|
assert original == decompressed
|
|
|
|
|
|
@pytest.mark.parametrize("quality", [1, 6, 9, 11])
|
|
@pytest.mark.parametrize("text_name", _test_utils.gather_text_inputs())
|
|
def test_multiple_process_and_flush(quality, text_name):
|
|
original = _test_utils.take_input(text_name)
|
|
chunk_size = 2048
|
|
chunks = _test_utils.chunk_input(original, chunk_size)
|
|
compressor = brotli.Compressor(quality=quality)
|
|
compressed = b''
|
|
for chunk in chunks:
|
|
compressed += compressor.process(chunk)
|
|
compressed += compressor.flush()
|
|
compressed += compressor.finish()
|
|
decompressed = brotli.decompress(compressed)
|
|
assert original == decompressed
|