Merge pull request #9 from pablosanjose/singlekwdef

Deal with (x; y=0) -> ... pattern

Author: oxinabox

src/function.jl

@@ -102,7 +102,11 @@ function splitdef(ex::Expr; throw::Bool=true)
             if !haskey(def, :args)
                 def[:args] = [arg]
             elseif !haskey(def, :kwargs)
-                def[:kwargs] = [arg]
+                if arg isa Expr && arg.head == :(=)
+                    def[:kwargs] = [:($(Expr(:kw, arg.args...)))]
+                else
+                    def[:kwargs] = [arg]
+                end
             else
                 return invalid_def("an invalid block expression as arguments")
             end

test/function.jl

@@ -545,6 +545,51 @@ function_form(short::Bool) = string(short ? "short" : "long", "-form")
             @test strip_lineno!(c_expr) == strip_lineno!(expr)
         end
 
+        @testset "(x; y = 0)" begin
+            f, expr = if short
+                @audit (x; y = 0) -> (x, y)
+            else
+                @audit function (x; y = 0); (x, y) end
+            end
+            @test length(methods(f)) == 1
+            @test f(0) == (0, 0)
+            @test f(0, y=1) == (0, 1)
+
+            # Note: the semi-colon is missing from the expression
+            d = splitdef(expr)
+            @test keys(d) == Set([:head, :args, :kwargs, :body])
+            @test d[:args] == [:x]
+            @test d[:kwargs] == [Expr(:kw, :y, 0)]
+
+            c_expr = combinedef(d)
+            expr = Expr(:->, Expr(:tuple, Expr(:parameters, Expr(:kw, :y, 0)), :x), Expr(:block, :((x, y))))
+            expr.head = short ? :-> : :function
+            @test strip_lineno!(c_expr) == strip_lineno!(expr)
+        end
+
+        @testset "(x; y = 0, _...)" begin
+            f, expr = if short
+                @audit (x; y = 0, _...) -> (x, y)
+            else
+                @audit function (x; y = 0, _...); (x, y) end
+            end
+            @test length(methods(f)) == 1
+            @test f(0) == (0, 0)
+            @test f(0, y=1) == (0, 1)
+            @test f(0, y=1, z=2) == (0, 1)
+
+            # Note: the semi-colon is missing from the expression
+            d = splitdef(expr)
+            @test keys(d) == Set([:head, :args, :kwargs, :body])
+            @test d[:args] == [:x]
+            @test d[:kwargs] == [Expr(:kw, :y, 0), :(_...)]
+
+            c_expr = combinedef(d)
+            expr = Expr(:->, Expr(:tuple, Expr(:parameters, Expr(:kw, :y, 0), :(_...)), :x), Expr(:block, :((x, y))))
+            expr.head = short ? :-> : :function
+            @test strip_lineno!(c_expr) == strip_lineno!(expr)
+        end
+
         @testset "Expr(:block, :x, :y)" begin
             expr = Expr(:->, Expr(:block, :x, :y), Expr(:block, :((x, y))))
             expr.head = short ? :-> : :function