Merge pull request #24 from oschulz/composed_function

Support Base.ComposedFunction

Author: DhairyaLGandhi

src/functor.jl

@@ -7,6 +7,10 @@ functor(::Type{<:NamedTuple}, x) = x, y -> y
 functor(::Type{<:AbstractArray}, x) = x, y -> y
 functor(::Type{<:AbstractArray{<:Number}}, x) = (), _ -> x
 
+@static if VERSION >= v"1.6"
+  functor(::Type{<:Base.ComposedFunction}, x) = (outer = x.outer, inner = x.inner), y -> Base.ComposedFunction(y.outer, y.inner)
+end
+
 function makefunctor(m::Module, T, fs = fieldnames(T))
   yᵢ = 0
   escargs = map(fieldnames(T)) do f

test/basics.jl

@@ -23,6 +23,16 @@ struct NoChildren
   y
 end
 
+@static if VERSION >= v"1.6"
+  @testset "ComposedFunction" begin
+    f1 = Foo(1.1, 2.2)
+    f2 = Bar(3.3)
+    @test Functors.functor(f1 ∘ f2)[1] == (outer = f1, inner = f2)
+    @test Functors.functor(f1 ∘ f2)[2]((outer = f1, inner = f2)) == f1 ∘ f2
+    @test fmap(x -> x + 10, f1 ∘ f2) == Foo(11.1, 12.2) ∘ Bar(13.3)
+  end
+end
+
 @testset "Nested" begin
   model = Bar(Foo(1, [1, 2, 3]))