Taylor expand equations

JuliaSymbolics · Oct 8, 2024 · c7c68b7 · c7c68b7
1 parent 92dfd5f
commit c7c68b7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/taylor.jl b/src/taylor.jl
@@ -56,3 +56,6 @@ function taylor(f, x, x0, n; kwargs...)
     # 3) substitute back x = x′ + x0
     return substitute(s, x′ => x - x0)
 end
+function taylor(eq::Equation, args...; kwargs...)
+    return taylor(eq.lhs, args...; kwargs...) ~ taylor(eq.rhs, args...; kwargs...)
+end
diff --git a/test/taylor.jl b/test/taylor.jl
@@ -26,3 +26,8 @@ end
 
 # around x ≠ 0
 @test substitute(taylor(√(x), x, 1, 0:6), x => x + 1) - taylor(√(1+x), x, 0:6) == 0
+
+# equations
+eq = sin(2*x) ~ 2*sin(x)*cos(x)
+eq = taylor(eq, x, 0:7)
+@test eq.lhs - eq.rhs == 0