now ...^(1//2) matches in the rule with sqrt, and ℯ^... matches in the rule with exp

Author: Bumblebee00

src/matchers.jl

@@ -194,6 +194,46 @@ function term_matcher_constructor(term, acSets)
             return nothing
         end
         return commutative_term_matcher
+    # if the operation is sqrt, we have to match also ^(1//2)
+    elseif operation(term)==sqrt
+        function sqrt_matcher(success, data, bindings)
+            !islist(data) && return nothing # if data is not a list, return nothing
+            data = car(data)
+            !iscall(data) && return nothing # if first element is not a call, return nothing
+            
+            # do the normal matcher
+            result = loop(data, bindings, matchers)
+            result !== nothing && return success(result, 1)
+
+            if (operation(data) === ^) && (arguments(data)[2] === 1//2)
+                T = symtype(arguments(data)[1])
+                frankestein = Term{T}(sqrt,[arguments(data)[1]])
+                result = loop(frankestein, bindings, matchers)
+                result !== nothing && return success(result, 1)
+            end
+            return nothing
+        end
+        return sqrt_matcher
+    # if the operation is exp, we have to match also ℯ^
+    elseif operation(term)==exp
+        function exp_matcher(success, data, bindings)
+            !islist(data) && return nothing # if data is not a list, return nothing
+            data = car(data)
+            !iscall(data) && return nothing # if first element is not a call, return nothing
+            
+            # do the normal matcher
+            result = loop(data, bindings, matchers)
+            result !== nothing && return success(result, 1)
+
+            if (operation(data) === ^) && (arguments(data)[1] === ℯ)
+                T = symtype(arguments(data)[2])
+                frankestein = Term{T}(exp,[arguments(data)[2]])
+                result = loop(frankestein, bindings, matchers)
+                result !== nothing && return success(result, 1)
+            end
+            return nothing
+        end
+        return exp_matcher
     else
         function term_matcher(success, data, bindings)
             !islist(data) && return nothing # if data is not a list, return nothing

test/rewrite.jl

@@ -113,7 +113,7 @@ end
     @test r_predicate(x+2) === (x, 2)
     @test r_predicate(x+2.0) !== (x, 2.0)
     # Note: r_predicate(x+2.0) doesnt return nothing, but (x+2.0, 0)
-    # becasue of the defslot
+    # because of the defslot
 end
 
 @testset "power matcher with negative exponent" begin
@@ -159,6 +159,16 @@ end
     @test r1(sqrt(a)) === (a, 1//2) # uses sqrt_matcher
 end
 
+@testset "Alternate form of special functions" begin
+    rsqrt = @rule sqrt(~x) => ~x
+    @test rsqrt(sqrt(x))===x
+    @test rsqrt((x)^(1//2))===x
+
+    rexp = @rule exp(~x) => ~x
+    @test rexp(exp(x)) === x
+    @test rexp(ℯ^x) === x
+end
+
 using SymbolicUtils: @capture
 
 @testset "Capture form" begin