Allow isdef() to match functions with return types

Author: kmsquire

src/utils.jl

@@ -216,7 +216,9 @@ end
 
 "Test for function definition expressions."
 isdef(ex) = ismatch(or_(:(function _(__) _ end),
-                        :(f_(__) = _)),
+                        :(function _(__)::_ _ end),
+                        :(f_(__) = _),
+                        :(f_(__)::_ = _)),
                     ex)
 
 isshortdef(ex) = (@capture(ex, (fcall_ = body_)) &&

test/runtests.jl

@@ -6,5 +6,6 @@ using Test
 include("match.jl")
 include("split.jl")
 include("destruct.jl")
+include("utils.jl")
 
 end

test/utils.jl

@@ -0,0 +1,12 @@
+using MacroTools: isdef
+
+@testset "utils" begin
+    ex1 = :(function foo(a) return a; end)
+    @test isdef(ex1)
+    ex2 = :(function bar(a)::Int return 1; end)
+    @test isdef(ex2)
+    ex3 = :(f(a) = a)
+    @test isdef(ex3)
+    ex4 = :(f(a)::Int = 1)
+    @test isdef(ex4)
+end