refactor: move tests and use modules to prevent spillover

Author: avik-pal

test/Project.toml

@@ -4,11 +4,13 @@ Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
+LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
 MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reactant_jll = "0192cb87-2b54-54ad-80e0-3be72ad8a3c0"
+SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

test/nn.jl renamed to test/nn/flux.jl

@@ -1,4 +1,4 @@
-using Reactant
+using Reactant, SafeTestsets, Test
 
 # parse some command-line arguments
 function extract_flag!(args, flag, default=nothing; typ=typeof(default))
@@ -39,13 +39,20 @@ if do_gpu_list
     # TODO set which gpu
 end
 
-include("layout.jl")
-include("tracing.jl")
-include("basic.jl")
-include("bcast.jl")
-include("struct.jl")
-include("closure.jl")
-include("compile.jl")
-include("buffer_donation.jl")
-include("nn.jl")
-include("nn_lux.jl")
+@testset "Reactant.jl Tests" begin
+    @safetestset "Layout" include("layout.jl")
+    @safetestset "Tracing" include("tracing.jl")
+    @safetestset "Basic" include("basic.jl")
+    @safetestset "Broadcast" include("bcast.jl")
+    @safetestset "Struct" include("struct.jl")
+    @safetestset "Closure" include("closure.jl")
+    @safetestset "Compile" include("compile.jl")
+    @safetestset "Buffer Donation" include("buffer_donation.jl")
+
+    @testset "Neural Networks" begin
+        @safetestset "NNlib Primitives" include("nn/nnlib.jl")
+        @safetestset "Flux.jl Integration" include("nn/flux.jl")
+        @safetestset "LuxLib Primitives" include("nn/luxlib.jl")
+        @safetestset "Lux Integration" include("nn/lux.jl")
+    end
+end