add test from issue 19

Author: MasonProtter

Project.toml

@@ -1,7 +1,7 @@
 name = "Mjolnir"
 uuid = "1154507a-7ac2-44fe-9f15-34617bca9db5"
 authors = ["Mike J Innes <[email protected]>"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 IRTools = "7869d1d1-7146-5819-86e3-90919afe41df"

test/trace.jl

@@ -164,3 +164,15 @@ tr = @trace f(::Matrix{Int32})
 f(xs) = sum(xs, dims = 1)
 tr = @trace f(::Matrix{Int32})
 @test returntype(tr) == Matrix{Int32}
+
+
+function negsquare(x)
+    if x > 0
+        return x^2
+    else
+        return -x^2
+    end
+end
+
+tr = @trace negsquare(::Float64)
+@test returntype(tr) == Float64