added identity function for pipes

Author: femtomc

src/CodeInfoTools.jl

@@ -9,7 +9,7 @@ import Base: show
 ##### Exports
 #####
 
-export var, Variable, Canvas, renumber, code_info, finish
+export var, Variable, Canvas, renumber, code_info, finish, get_slot
 
 #####
 ##### Utilities
@@ -28,7 +28,7 @@ function code_info(f, tt; generated=true, debuginfo=:default)
 end
 
 function get_slot(ci::CodeInfo, s::Symbol)
-    ind = findfirst(ci.slotnames, s)
+    ind = findfirst(el -> el == s, ci.slotnames)
     ind == nothing && return 
     return Core.Compiler.SlotNumber(ind)
 end

test/misc.jl

@@ -0,0 +1,17 @@
+
+@testset "Base.:(+) -- SSAValues" begin
+    @test (+)(Core.SSAValue(1), 1) == Core.SSAValue(2)
+    @test (+)(1, Core.SSAValue(1)) == Core.SSAValue(2)
+end
+
+@testset "Pipe -- misc." begin
+    ir = code_info(g, Tuple{Int})
+    p = CodeInfoTools.Pipe(ir)
+    for (v, st) in p
+    end
+    display(p)
+    display(length(p))
+    p = CodeInfoTools.Pipe(ir)
+    p = identity(p)
+    get_slot(p, :x)
+end

test/pipe.jl

@@ -144,16 +144,3 @@ end
     display(finish(p))
 end
 
-@testset "Base.:(+) -- SSAValues" begin
-    @test (+)(Core.SSAValue(1), 1) == Core.SSAValue(2)
-    @test (+)(1, Core.SSAValue(1)) == Core.SSAValue(2)
-end
-
-@testset "Pipe -- misc." begin
-    ir = code_info(g, Tuple{Int})
-    p = CodeInfoTools.Pipe(ir)
-    for (v, st) in p
-    end
-    display(p)
-    display(length(p))
-end

test/runtests.jl

@@ -5,5 +5,6 @@ using Test
 
 include("canvas.jl")
 include("pipe.jl")
+include("misc.jl")
 
 end # module