/tests/web/test_utils.py

https://bitbucket.org/prologic/circuits/ · Python · 25 lines · 18 code · 6 blank · 1 comment · 2 complexity · 4c48b02667c89e2ce7c4f7e74fbbe717 MD5 · raw file

  1. #!/usr/bin/env python
  2. from io import BytesIO
  3. try:
  4. from gzip import decompress
  5. except ImportError:
  6. import zlib
  7. decompress = zlib.decompressobj(16+zlib.MAX_WBITS).decompress # NOQA
  8. from circuits.web.utils import compress
  9. from circuits.web.utils import get_ranges
  10. def test_ranges():
  11. assert get_ranges("bytes=3-6", 8) == [(3, 7)]
  12. assert get_ranges("bytes=2-4,-1", 8) == [(2, 5), (7, 8)]
  13. def test_gzip():
  14. s = b"Hello World!"
  15. contents = BytesIO(s)
  16. compressed = b"".join(compress(contents, 1))
  17. uncompressed = decompress(compressed)
  18. assert uncompressed == s
  19. contents.close()