Fix some small errors

Author: rhpvorderman

src/isal/igzip.py

@@ -31,7 +31,7 @@
 
 from . import isal_zlib
 
-__all__ = ["IGzipFile", "open", "compress", "decompress"]
+__all__ = ["IGzipFile", "open", "compress", "decompress", "BadGzipFile"]
 
 _COMPRESS_LEVEL_FAST = isal_zlib.ISAL_BEST_SPEED
 _COMPRESS_LEVEL_TRADEOFF = isal_zlib.ISAL_DEFAULT_COMPRESSION
@@ -40,6 +40,9 @@
 
 BUFFER_SIZE = _compression.BUFFER_SIZE
 
+class BadGzipFile(OSError):
+    pass
+
 
 # The open method was copied from the python source with minor adjustments.
 def open(filename, mode="rb", compresslevel=_COMPRESS_LEVEL_TRADEOFF,

tests/test_gzip_compliance.py

@@ -456,7 +456,7 @@ def test_zero_padded_file(self):
     def test_igzip_BadGzipFile_exception(self):
         self.assertTrue(issubclass(igzip.BadGzipFile, OSError))
 
-    def test_bad_igzip_file(self):
+    def test_bad_gzip_file(self):
         with open(self.filename, 'wb') as file:
             file.write(data1 * 50)
         with igzip.IGzipFile(self.filename, 'r') as file:
@@ -555,7 +555,7 @@ def test_bytes_filename(self):
 
     def test_decompress_limited(self):
         """Decompressed data buffering should be limited"""
-        bomb = igzip.compress(b'\0' * int(2e6), compresslevel=9)
+        bomb = igzip.compress(b'\0' * int(2e6), compresslevel=3)
         self.assertLess(len(bomb), io.DEFAULT_BUFFER_SIZE)
 
         bomb = io.BytesIO(bomb)
@@ -569,7 +569,7 @@ def test_decompress_limited(self):
 
     def test_compress(self):
         for data in [data1, data2]:
-            for args in [(), (1,), (6,), (9,)]:
+            for args in [(), (1,), (2,), (3,), (0,)]:
                 datac = igzip.compress(data, *args)
                 self.assertEqual(type(datac), bytes)
                 with igzip.IGzipFile(fileobj=io.BytesIO(datac), mode="rb") as f: