added tests for gruntz demo

Author: anutosh491

integration_tests/gruntz_demo3.py

@@ -1,5 +1,5 @@
 from lpython import S, str
-from sympy import Symbol, Pow, sin, oo, pi, E, Mul, Add
+from sympy import Symbol, Pow, sin, oo, pi, E, Mul, Add, oo, log, exp, cos
 
 def mrv(e: S, x: S) -> list[S]:
     """
@@ -149,32 +149,32 @@ def leadterm(e: S, x: S) -> list[S]:
     Returns the leading term a*x**b as a list [a, b].
     """
     if e == sin(x)/x:
-        case1: list[S] = [S(1), S(0)]
-        return case1
+        l1: list[S] = [S(1), S(0)]
+        return l1
     elif e == S(2)*sin(x)/x:
-        case2: list[S] = [S(2), S(0)]
-        return case2
+        l2: list[S] = [S(2), S(0)]
+        return l2
     elif e == sin(S(2)*x)/x:
-        case3: list[S] = [S(2), S(0)]
-        return case3
+        l3: list[S] = [S(2), S(0)]
+        return l3
     elif e == sin(x)**S(2)/x:
-        case4: list[S] = [S(1), S(1)]
-        return case4
+        l4: list[S] = [S(1), S(1)]
+        return l4
     elif e == sin(x)/x**S(2):
-        case5: list[S] = [S(1), S(-1)]
-        return case5
+        l5: list[S] = [S(1), S(-1)]
+        return l5
     elif e == sin(x)**S(2)/x**S(2):
-        case6: list[S] = [S(1), S(0)]
-        return case6
+        l6: list[S] = [S(1), S(0)]
+        return l6
     elif e == sin(sin(sin(x)))/sin(x):
-        case7: list[S] = [S(1), S(0)]
-        return case7
+        l7: list[S] = [S(1), S(0)]
+        return l7
     elif e == S(2)*log(x+S(1))/x:
-        case8: list[S] = [S(2), S(0)]
-        return case8
+        l8: list[S] = [S(2), S(0)]
+        return l8
     elif e == sin((log(x+S(1))/x)*x)/x:
-        case9: list[S] = [S(1), S(0)]
-        return case9
+        l9: list[S] = [S(1), S(0)]
+        return l9
     raise NotImplementedError(f"Can't calculate the leadterm of {e}.")
 
 def mrv_leadterm(e: S, x: S) -> list[S]:
@@ -197,9 +197,9 @@ def mrv_leadterm(e: S, x: S) -> list[S]:
 
     """
 
-    #w = Dummy('w', real=True, positive=True)
-    #e = rewrite(e, x, w)
-    #return e.leadterm(w)
+    # w = Dummy('w', real=True, positive=True)
+    # e = rewrite(e, x, w)
+    # return e.leadterm(w)
     w: S = Symbol('w')
     newe: S = rewrite(e, x, w)
     coeff_exp_list: list[S] = leadterm(newe, w)
@@ -265,7 +265,24 @@ def gruntz(e: S, z: S, z0: S, dir: str ="+") -> S:
 # test
 def test():
     x: S = Symbol('x')
-    ans: S = gruntz(sin(x)/x, x, S(0), "+")
-    print(ans)
+    print(gruntz(sin(x)/x, x, S(0), "+"))
+    print(gruntz(S(2)*sin(x)/x, x, S(0), "+"))
+    print(gruntz(sin(S(2)*x)/x, x, S(0), "+"))
+    print(gruntz(sin(x)**S(2)/x, x, S(0), "+"))
+    print(gruntz(sin(x)/x**S(2), x, S(0), "+"))
+    print(gruntz(sin(x)**S(2)/x**S(2), x, S(0), "+"))
+    print(gruntz(sin(sin(sin(x)))/sin(x), x, S(0), "+"))
+    print(gruntz(S(2)*log(x+S(1))/x, x, S(0), "+"))
+    print(gruntz(sin((log(x+S(1))/x)*x)/x, x, S(0), "+"))
+
+    assert gruntz(sin(x)/x, x, S(0)) == S(1)
+    assert gruntz(S(2)*sin(x)/x, x, S(0)) == S(2)
+    assert gruntz(sin(S(2)*x)/x, x, S(0)) == S(2)
+    assert gruntz(sin(x)**S(2)/x, x, S(0)) == S(0)
+    assert gruntz(sin(x)/x**S(2), x, S(0)) == oo
+    assert gruntz(sin(x)**S(2)/x**S(2), x, S(0)) == S(1)
+    assert gruntz(sin(sin(sin(x)))/sin(x), x, S(0)) == S(1)
+    assert gruntz(S(2)*log(x+S(1))/x, x, S(0)) == S(2)
+    assert gruntz(sin((log(x+S(1))/x)*x)/x, x, S(0)) == S(1)
 
 test()