Adapt to init_worker_code

Author: christiangnrd

test/runtests.jl

@@ -1,9 +1,13 @@
+using Pkg
+Pkg.add(url="https://github.com/christiangnrd/ParallelTestRunner.jl", rev="total_time")
+
 using ParallelTestRunner: runtests, parse_args
 import GPUArrays
 
 include("testsuite.jl")
 
-const init_code = quote
+# const init_code = quote
+const init_worker_code = quote
     using Test, JLArrays, SparseArrays
 
     include("testsuite.jl")
@@ -18,11 +22,17 @@ const init_code = quote
     end
 end
 
+const init_code = quote
+    using Test, JLArrays, SparseArrays
+
+    import Main: TestSuite
+end
+
 args = parse_args(ARGS)
 
 testsuite = Dict{String, Expr}()
 for AT in (:JLArray, :Array), name in keys(TestSuite.tests)
     testsuite["$(AT)/$name"] = :(TestSuite.tests[$name]($AT))
 end
 
-runtests(GPUArrays, ARGS; init_code, testsuite)
+runtests(GPUArrays, ARGS; init_code, init_worker_code, testsuite)