aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/plot.py
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2019-12-23 07:32:18 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2019-12-23 07:32:18 +0300
commit0d25b5d6957d76bab486e2279c99e130c19d5c31 (patch)
treec8c7d6fe4766470bda03e49a004afaa03c967b16 /plot.py
parentTravis: add badge to README (diff)
downloadsorting-algorithms-0d25b5d6957d76bab486e2279c99e130c19d5c31.tar.gz
sorting-algorithms-0d25b5d6957d76bab486e2279c99e130c19d5c31.zip
pylint/pep8 fixes
Diffstat (limited to 'plot.py')
-rw-r--r--plot.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/plot.py b/plot.py
index 55a1d6f..0a8ec75 100644
--- a/plot.py
+++ b/plot.py
@@ -10,11 +10,13 @@ from algorithms.input_kind import InputKind
from algorithms.params import AlgorithmParameters
import algorithms.registry as registry
+
_DEFAULT_ITERATIONS = 100
_DEFAULT_INPUT_KIND = InputKind.AVERAGE
_DEFAULT_MIN_LENGTH = 0
_DEFAULT_MAX_LENGTH = 200
+
def plot_algorithm(algorithm, input_kind=_DEFAULT_INPUT_KIND,
min_len=_DEFAULT_MIN_LENGTH,
max_len=_DEFAULT_MAX_LENGTH,
@@ -29,6 +31,7 @@ def plot_algorithm(algorithm, input_kind=_DEFAULT_INPUT_KIND,
iterations=iterations)
params.plot_running_time(output_path)
+
def _parse_non_negative_integer(s):
try:
n = int(s)
@@ -38,6 +41,7 @@ def _parse_non_negative_integer(s):
raise argparse.ArgumentTypeError('must be a non-negative integer')
return n
+
def _parse_positive_integer(s):
try:
n = int(s)
@@ -47,28 +51,34 @@ def _parse_positive_integer(s):
raise argparse.ArgumentTypeError('must be a positive 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:]
@@ -98,8 +108,10 @@ def _parse_args(args=None):
return parser.parse_args(args)
+
def main(args=None):
plot_algorithm(**vars(_parse_args(args)))
+
if __name__ == '__main__':
main()