diff options
Diffstat (limited to 'test/toolkit.py')
| -rw-r--r-- | test/toolkit.py | 19 |
1 files changed, 8 insertions, 11 deletions
diff --git a/test/toolkit.py b/test/toolkit.py index b6c3cd3..d14ce23 100644 --- a/test/toolkit.py +++ b/test/toolkit.py @@ -135,32 +135,29 @@ class Tools: return args @staticmethod - def _block_settings_to_args(algorithm, mode, use_boxes=False): - args = [ + def _block_settings_to_args(algorithm, mode): + return [ "--algorithm", str(algorithm), "--mode", str(mode), ] - if use_boxes: - args.append("--use-boxes") - return args @staticmethod - def _build_block_args(algorithm, mode, inputs, use_boxes=False): - args = Tools._block_settings_to_args(algorithm, mode, use_boxes) + def _build_block_args(algorithm, mode, inputs): + args = Tools._block_settings_to_args(algorithm, mode) if isinstance(inputs, Iterable): args.extend(Tools._block_inputs_to_args(iter(inputs))) else: args.extend(inputs.to_args()) return args - def run_encrypt_block(self, algorithm, mode, inputs, use_boxes=False): - args = self._build_block_args(algorithm, mode, inputs, use_boxes) + def run_encrypt_block(self, algorithm, mode, inputs): + args = self._build_block_args(algorithm, mode, inputs) return self.run(self._ENCRYPT_BLOCK, args) - def run_decrypt_block(self, algorithm, mode, inputs, use_boxes=False): - args = self._build_block_args(algorithm, mode, inputs, use_boxes) + def run_decrypt_block(self, algorithm, mode, inputs): + args = self._build_block_args(algorithm, mode, inputs) return self.run(self._DECRYPT_BLOCK, args) @staticmethod |