aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--challenge-031/lubos-kolouch/python/ch_1.py49
1 files changed, 49 insertions, 0 deletions
diff --git a/challenge-031/lubos-kolouch/python/ch_1.py b/challenge-031/lubos-kolouch/python/ch_1.py
new file mode 100644
index 0000000000..c507c163ec
--- /dev/null
+++ b/challenge-031/lubos-kolouch/python/ch_1.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+import sys
+import argparse
+import pytest
+
+def get_args():
+ parser = argparse.ArgumentParser(\
+ description='Divisor check')
+ parser.add_argument(\
+ '-n',
+ '--nominator',
+ type=int,
+ help='Nominator',
+ required=True)
+ parser.add_argument(\
+ '-d',
+ '--denominator',
+ type=int,
+ help='Denominator',
+ required=True)
+ return parser.parse_args()
+
+def check_divide(nom,den):
+ try:
+ nom/den
+ except:
+ return("error")
+
+ return "ok"
+
+def main():
+ if sys.version_info[0] < 3:
+ print("ERROR: Python3 required.")
+ exit(1)
+ init_numbers = get_args()
+ print(check_divide(init_numbers.nominator, init_numbers.denominator))
+
+def test_div():
+ assert(check_divide(3,4)=="ok")
+ assert(check_divide(0,0)=="error")
+ assert(check_divide(1,0)=="error")
+ assert(check_divide(0,1)=="ok")
+
+main()
+
+test_div()
+#EOF