diff --git a/integration_tests/test_gruntz.py b/integration_tests/test_gruntz.py index 376f68495a..33b940f86f 100644 --- a/integration_tests/test_gruntz.py +++ b/integration_tests/test_gruntz.py @@ -1,5 +1,5 @@ from lpython import S -from sympy import Symbol +from sympy import Symbol, log def mmrv(e: S, x: S) -> list[S]: if not e.has(x): @@ -8,6 +8,10 @@ def mmrv(e: S, x: S) -> list[S]: elif e == x: list1: list[S] = [x] return list1 + elif e.func == log: + arg0: S = e.args[0] + list2: list[S] = mmrv(arg0, x) + return list2 else: raise @@ -26,4 +30,11 @@ def test_mrv(): assert ele1 == x assert len(ans2) == 1 + # Case 3 + ans3: list[S] = mmrv(log(x), x) + ele2: S = ans3[0] + print(ele2) + assert ele2 == x + assert len(ans2) == 1 + test_mrv() \ No newline at end of file