Merge pull request #542 from raphaelchinchilla/return_tuples

Included dispatch to toexpr that creates Tuples correctly
JuliaSymbolics · Aug 20, 2023 · e4812a3 · e4812a3
2 parents 91879dd + 564ba54
commit e4812a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/code.jl b/src/code.jl
@@ -188,6 +188,10 @@ function toexpr(O, st)
     end
 end
 
+function toexpr(O::Tuple, st)
+    :(($(toexpr.(O, (st,))...),))
+end
+
 # Call elements of vector arguments by their name.
 @matchable struct DestructuredArgs
     elems

diff --git a/test/code.jl b/test/code.jl
@@ -183,6 +183,8 @@ nanmath_st.rewrites[:nanmath] = true
               :(SparseVector(10, $(spvec.nzind), [a])))
     test_repr(toexpr(MakeTuple((a, b, a+b))),
               :((a,b,$(+)(a,b))))
+    test_repr(toexpr((a, b, a+b)),
+              :((a,b,$(+)(a,b))))
 
     @test SpawnFetch{Multithreaded}([()->1,()->2],vcat)|>toexpr|>eval == [1,2]
     @test @elapsed(SpawnFetch{Multithreaded}([:(()->sleep(2)),