From da61bd1bc09f726f3821b2cdd2376e4fc677a7de Mon Sep 17 00:00:00 2001 From: Philipp Gesang Date: Fri, 7 Apr 2017 11:15:00 +0200 Subject: [PATCH] allow test_compression_level.py as module --- testing/test_compression_level.py | 21 ++++++++++++++++++++- 1 files changed, 20 insertions(+), 1 deletions(-) diff --git a/testing/test_compression_level.py b/testing/test_compression_level.py index 7bbebf9..69880be 100644 --- 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 -- 1.7.1