Skip to content

Commit 0c5dc48

Browse files
committed
Update
1 parent e740753 commit 0c5dc48

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/functions.jl

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -926,14 +926,21 @@ function _is_approx(x::AbstractArray, y::AbstractArray; kwargs...)
926926
all(z -> _is_approx(z[1], z[2]; kwargs...), zip(x, y))
927927
end
928928

929+
function _is_univariate_plus(f)
930+
if f.head == :+ && length(f.args) == 1
931+
return only(f.args) isa ScalarNonlinearFunction
932+
end
933+
return false
934+
end
935+
929936
function Base.isapprox(
930937
f::ScalarNonlinearFunction,
931938
g::ScalarNonlinearFunction;
932939
kwargs...,
933940
)
934-
if f.head == :+ && length(f.args) == 1
935-
return isapprox(only(f.args), g; kwargs...)
936-
elseif g.head == :+ && length(g.args) == 1
941+
if _is_univariate_plus(f)
942+
return isapprox(only(f.args), g.args; kwargs...)
943+
elseif _is_univariate_plus(g)
937944
return isapprox(f, only(g.args); kwargs...)
938945
end
939946
if f.head != g.head || length(f.args) != length(g.args)

0 commit comments

Comments
 (0)