Merge pull request #82 from jverzani/issue_sympy_554

close JuliaPy/SymPy.jl#554
jverzani · Jan 17, 2025 · 10d12cb · 10d12cb · jverzani · Jan 17, 2025
2 parents fba40ef + 59d9a9a
commit 10d12cb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SymPyCore"
 uuid = "458b697b-88f0-4a86-b56b-78b75cfb3531"
 authors = ["jverzani <jverzani@gmail.com> and contributors"]
-version = "0.2.8"
+version = "0.2.9"
 
 [deps]
 CommonEq = "3709ef60-1bee-4518-9f2f-acd86f176c50"

diff --git a/src/lambdify.jl b/src/lambdify.jl
@@ -191,7 +191,9 @@ sympy_fn_julia_fn = Dict(
 
 const  fn_map = Dict(k => last(v) for (k,v) ∈ pairs(sympy_fn_julia_fn))
 
-map_fn(key, fn_map) = haskey(fn_map, key) ? fn_map[key] : Symbol(key)
+map_fn(key, fn_map) = haskey(fn_map, key) ? fn_map[key] :
+    isdefined(@__MODULE__, Symbol(key)) ? Symbol(key) :
+    error("Lambdify doesn't know what to do with $key. Sorry.")
 
 ##
 
@@ -272,8 +274,10 @@ function walk_expression(ex; values=Dict(), fns=Dict())
         return vals_map[fn]
     end
 
+    fn′ = map_fn(fn, fns_map)
+
     as = args(ex)
-    Expr(:call, map_fn(fn, fns_map), [walk_expression(a, values=values, fns=fns) for a in as]...)
+    Expr(:call, fn′, [walk_expression(a, values=values, fns=fns) for a in as]...)
 end
 
 """