Merge pull request #844 from kolcon/master

Snake solution 1

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