From da33d2064c00b355a933f52a200020af813a4527 Mon Sep 17 00:00:00 2001 From: Brandon Rodriguez <brodriguez8774@gmail.com> Date: Thu, 5 Dec 2019 13:27:38 -0500 Subject: [PATCH] Add ability for user to select algorithm --- main.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 3745b43..1a0453b 100644 --- a/main.py +++ b/main.py @@ -337,7 +337,14 @@ def simplex(): if __name__ == '__main__': logger.info('Starting program.') - ford_fulkerson() - simplex() + logger.info('Choose an option:') + logger.info(' 1) Solve Maximum Flow via Ford Fulkerson.') + logger.info(' 2) Solve Maximum Flow via Simplex.') + + user_input = str(input()).strip() + if user_input == '1': + ford_fulkerson() + elif user_input == '2': + simplex() logger.info('Terminating program.') -- GitLab