Automatically unwrap and wrap Nums for rule application (#1176)

JuliaSymbolics · Jun 23, 2024 · e4c3281 · e4c3281
1 parent 17ae756
commit e4c3281
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/rewrite-helpers.jl b/src/rewrite-helpers.jl
@@ -121,6 +121,8 @@ function filterchildren!(r::Any, y, acc)
     end
 end
 
+(f::SymbolicUtils.Rule)(x::Num) = wrap(f(unwrap(x)))
+
 module RewriteHelpers
 import Symbolics: replacenode, hasnode, filterchildren, unwrap
 export replacenode, hasnode, filterchildren, unwrap

diff --git a/test/rewrite_helpers.jl b/test/rewrite_helpers.jl
@@ -115,3 +115,9 @@ let
     @test isequal(filterchildren(is_derivative, ex4), [D(Y), D(my_f(1,Z))])
 end
 
+# https://github.com/JuliaSymbolics/Symbolics.jl/issues/1175
+let
+    @variables w z α::Real β::Real;
+    r3 = @rule ~x * +(~~ys) => sum(map(y-> ~x * y, ~~ys));
+    @test r3(2 * (w+w+α+β)) isa Num
+end