diff --git a/tests/missing_colon.py b/tests/missing_colon.py index 20edef5..ec220af 100755 --- a/tests/missing_colon.py +++ b/tests/missing_colon.py @@ -1,10 +1,15 @@ #!/usr/bin/env python3 -def division(a: float, b: float) -> float - return a/b +def division(a: float, b: float) -> float: + if b == 0: + raise ValueError("Cannot divide by zero") + return a / b if __name__ == "__main__": - print(division(123, 15)) + try: + print(division(23, 0)) + except ValueError as e: + print(e)