From a84cbf18f18726535e10696cccc62a13fce507d1 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Tue, 28 Nov 2023 23:54:38 +0100 Subject: test.py: formatting fixes --- test.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/test.py b/test.py index cbf8880..b03bf88 100755 --- a/test.py +++ b/test.py @@ -12,9 +12,11 @@ import sys from algorithms.input_kind import InputKind import algorithms.registry as registry + _DEFAULT_INPUT_KIND = InputKind.AVERAGE _DEFAULT_LENGTH = 100 + def test(algorithm, input_kind=_DEFAULT_INPUT_KIND, length=_DEFAULT_LENGTH): if isinstance(algorithm, str): algorithm = registry.get(algorithm) @@ -24,6 +26,7 @@ def test(algorithm, input_kind=_DEFAULT_INPUT_KIND, length=_DEFAULT_LENGTH): output = output.tolist() print(output) + def _parse_non_negative_integer(s): try: n = int(s) @@ -33,28 +36,34 @@ def _parse_non_negative_integer(s): raise argparse.ArgumentTypeError('must be a non-negative integer') return n + def _parse_input_kind(s): try: return InputKind(s) except ValueError: raise argparse.ArgumentTypeError('invalid input kind: ' + str(s)) + def _format_algorithm(codename): return '* {}: {}'.format(codename, registry.get(codename).display_name) + def _format_available_algorithms(): descr = 'available algorithms (in the CODENAME: DISPLAY_NAME format):\n' return descr + '\n'.join(map( _format_algorithm, sorted(registry.get_codenames()))) + def _format_description(): return _format_available_algorithms() + def _create_argument_parser(): return argparse.ArgumentParser( description=_format_description(), formatter_class=argparse.RawDescriptionHelpFormatter) + def _parse_args(args=None): if args is None: args = sys.argv[1:] @@ -75,8 +84,10 @@ def _parse_args(args=None): return parser.parse_args(args) + def main(args=None): test(**vars(_parse_args(args))) + if __name__ == '__main__': main() -- cgit v1.2.3