allow test_compression_level.py as module

diff --git a/testing/test_compression_level.py b/testing/test_compression_level.py
index 7bbebf9..69880be 100644 (file)
--- a/testing/test_compression_level.py
+++ b/testing/test_compression_level.py
@@ -25,6 +25,7 @@ import unittest
 from . import run_benchmark
 from . import create_pseudo_random_files
 
+DEFAULTS_COMPRESS_LEVEL_TEST = (int(42 / 10), 42, 42, 1, 8)
 
 class CompressLevelTest(unittest.TestCase):
     """
@@ -35,7 +36,25 @@ class CompressLevelTest(unittest.TestCase):
 
     BUF_SIZE = 2**12
 
-    def __init__(self, num_dir, num_files, size, compression_lvl_1, compression_lvl_2):
+    def __init__(self, num_dir=None, num_files=None, size=None,
+                 compression_lvl_1=None, compression_lvl_2=None):
+        if isinstance (num_dir, str) and \
+           num_files         is None and \
+           size              is None and \
+           compression_lvl_1 is None and \
+           compression_lvl_2 is None:
+            # invoked as module
+            num_dir           = DEFAULTS_COMPRESS_LEVEL_TEST [0]
+            num_files         = DEFAULTS_COMPRESS_LEVEL_TEST [1]
+            size              = DEFAULTS_COMPRESS_LEVEL_TEST [2]
+            compression_lvl_1 = DEFAULTS_COMPRESS_LEVEL_TEST [3]
+            compression_lvl_2 = DEFAULTS_COMPRESS_LEVEL_TEST [4]
+        elif num_dir           is None or \
+             num_files         is None or \
+             size              is None or \
+             compression_lvl_1 is None or \
+             compression_lvl_2 is None:
+            raise ValueError
         super(CompressLevelTest, self).__init__('test_compression_level')
         self.num_dir = num_dir
         self.num_files = num_files