From 0f9c3c04b5765ee879cd8ffeac26eb91e015fba8 Mon Sep 17 00:00:00 2001 From: Lubos Kolouch Date: Sun, 27 Oct 2019 09:58:38 +0100 Subject: Snake solution 1 --- challenge-031/lubos-kolouch/python/ch_1.py | 49 ++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 challenge-031/lubos-kolouch/python/ch_1.py 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 -- cgit