Add tests

Author: vchuravy

test/devices.jl

@@ -0,0 +1,14 @@
+function devices_testsuite(Backend)
+    backend = Backend()
+
+    current_device = KernelAbstractions.device!(backend)
+    for i in KernelAbstractions.ndevices(backend)
+        KernelAbstractions.device!(backend, i)
+        @test KernelAbstractions.device(backend) == i
+    end
+
+    @test_throws ArgumentError KernelAbstractions.device!(backend, 0)
+    @test_throws ArgumentError KernelAbstractions.device!(backend, KernelAbstractions.ndevices(backend)+1)
+
+    KernelAbstractions.device!(backend, current_device)
+end

test/testsuite.jl

@@ -31,6 +31,7 @@ include("private.jl")
 include("unroll.jl")
 include("nditeration.jl")
 include("copyto.jl")
+include("devices.jl")
 include("print_test.jl")
 include("compiler.jl")
 include("reflection.jl")
@@ -67,6 +68,10 @@ function testsuite(backend, backend_str, backend_mod, AT, DAT; skip_tests = Set{
         copyto_testsuite(backend, AT)
     end
 
+    @conditional_testset "Devices" skip_tests begin
+        devices_testsuite(backend)
+    end
+
     @conditional_testset "Printing" skip_tests begin
         printing_testsuite(backend)
     end