use abs in gcd and lcm

Author: Smit-create

integration_tests/test_math.py

@@ -71,13 +71,17 @@ def test_gcd():
     assert i == 2
     i = gcd(21, 14)
     assert i == 7
+    i = gcd(21, -12)
+    assert i == 3
 
 def test_lcm():
     i: i32
     i = lcm(10, 4)
     assert i == 20
     i = lcm(21, 14)
     assert i == 42
+    i = lcm(21, -12)
+    assert i == 84
 
 
 test_factorial_1()

src/runtime/math.py

@@ -123,6 +123,10 @@ def gcd(a: i32, b: i32) -> i32:
     Returns greatest common divisor of `a` and `b`
     """
     temp: i32
+    if a < 0:
+        a = -a
+    if b < 0:
+        b = -b
     while b != 0:
         a = mod(a, b)
         temp = a
@@ -135,4 +139,10 @@ def lcm(a: i32, b: i32) -> i32:
     """
     Returns least common multiple of `a` and `b`
     """
+    if a < 0:
+        a = -a
+    if b < 0:
+        b = -b
+    if a*b == 0:
+        return 0
     return (a*b)//gcd(a, b)