fix: implement isequal_with_metadata for custom symbolics

Author: AayushSabharwal

src/types.jl

@@ -314,6 +314,21 @@ function isequal_with_metadata(a::BasicSymbolic{T}, b::BasicSymbolic{S})::Bool w
     _isequal(a, b, E; comparator = isequal_with_metadata)::Bool && isequal_with_metadata(metadata(a), metadata(b)) || return false
 end
 
+function isequal_with_metadata(a::Symbolic, b::Symbolic)::Bool
+    a === b && return true
+    typeof(a) == typeof(b) || return false
+
+    ma = metadata(a)
+    mb = metadata(b)
+    if iscall(a) && iscall(b)
+        return isequal_with_metadata(operation(a), operation(b)) && isequal_with_metadata(arguments(a), arguments(b)) && isequal_with_metadata(ma, mb)
+    elseif iscall(a) || iscall(b)
+        return false
+    else
+        return isequal_with_metadata(ma, mb)
+    end
+end
+
 """
     $(TYPEDSIGNATURES)
 

test/hash_consing.jl

@@ -1,5 +1,6 @@
 using SymbolicUtils, Test
-using SymbolicUtils: Term, Add, Mul, Div, Pow, hash2, metadata
+using SymbolicUtils: Term, Add, Mul, Div, Pow, hash2, metadata, Symbolic, isequal_with_metadata
+import TermInterface
 
 struct Ctx1 end
 struct Ctx2 end
@@ -147,3 +148,42 @@ end
         @test ex2.hash2[] != h
     end
 end
+
+struct MySymbolic <: Symbolic{Real}
+    sym::BasicSymbolic{Real}
+end
+
+TermInterface.iscall(x::MySymbolic) = iscall(x.sym)
+TermInterface.operation(x::MySymbolic) = operation(x.sym)
+TermInterface.arguments(x::MySymbolic) = arguments(x.sym)
+SymbolicUtils.metadata(x::MySymbolic) = metadata(x.sym)
+Base.isequal(a::MySymbolic, b::MySymbolic) = isequal(a.sym, b.sym)
+
+@testset "`isequal_with_metadata` on custom symbolics" begin
+    @syms x::Real
+    xx = setmetadata(x, Int, 3)
+    @test isequal(x, xx)
+    @test !isequal_with_metadata(x, xx)
+    myx = MySymbolic(x)
+    myxx = MySymbolic(xx)
+    @test isequal(myx, myxx)
+    @test !isequal_with_metadata(myx, myxx)
+
+    ex = 2x
+    exx = 2xx
+    myex = MySymbolic(ex)
+    myexx = MySymbolic(exx)
+    @test isequal(ex, exx)
+    @test !isequal_with_metadata(ex, exx)
+    @test isequal(myex, myexx)
+    @test !isequal_with_metadata(myex, myexx)
+
+    t = Term{Real}(max, Any[x, myex])
+    tt = Term{Real}(max, Any[xx, myexx])
+    @test isequal(t, tt)
+    @test !isequal_with_metadata(t, tt)
+    myt = MySymbolic(t)
+    mytt = MySymbolic(tt)
+    @test isequal(myt, mytt)
+    @test !isequal_with_metadata(myt, mytt)
+end