Implement Base.:(==) for thunks (#367)

JuliaDiff · Jun 9, 2021 · 7d667c5 · 7d667c5 · mzgubic · Jun 9, 2021
1 parent 6e2812e
commit 7d667c5
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "ChainRulesCore"
 uuid = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
-version = "0.10.2"
+version = "0.10.3"
 
 [deps]
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"

diff --git a/src/differentials/thunks.jl b/src/differentials/thunks.jl
@@ -13,6 +13,10 @@ end
     return element, (val, new_state)
 end
 
+Base.:(==)(a::AbstractThunk, b::AbstractThunk) = unthunk(a) == unthunk(b)
+Base.:(==)(a::AbstractThunk, b) = unthunk(a) == b
+Base.:(==)(a, b::AbstractThunk) = a == unthunk(b)
+
 """
     @thunk expr
 

diff --git a/test/differentials/thunks.jl b/test/differentials/thunks.jl
@@ -1,6 +1,12 @@
 @testset "Thunk" begin
     @test @thunk(3) isa Thunk
 
+    @testset "==" begin
+        @test @thunk(3.2) == InplaceableThunk(@thunk(3.2), x -> x + 3.2)
+        @test @thunk(3.2) == 3.2
+        @test 3.2 == InplaceableThunk(@thunk(3.2), x -> x + 3.2)
+    end
+
     @testset "show" begin
         rep = repr(Thunk(rand))
         @test occursin(r"Thunk\(.*rand.*\)", rep)