Added test for env in param construction

GuillaumeLam · Nov 23, 2021 · 658bba5 · 658bba5
1 parent 0d15d66
commit 658bba5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/lsm_tests.jl b/test/lsm_tests.jl
@@ -12,6 +12,13 @@ using CUDA
         @test isa(y, Vector{Float64})
     end
 
+    @testset "Params" begin
+        cartpole_params = LSM_Params(4,2,"cartpole")
+        default_params = LSM_Params(4,2)
+
+        @test cartpole_params == default_params
+    end
+
     @testset "Visual" begin
         params = LSM_Params(4,2)
         lsm = LSM(params, visual=true)