use reset! in RL test

Neuroblox · Feb 15, 2024 · efaf386 · efaf386
1 parent 183c428
commit efaf386
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/reinforcement_learning.jl b/test/reinforcement_learning.jl
@@ -69,4 +69,10 @@ using CSV
     @test any(init_params[map_idxs[idxs_weight]] .!= final_params[map_idxs[idxs_weight]])
     # All non-weight parameters need to be the same.
     @test all(init_params[map_idxs[idxs_other_params]] .== final_params[map_idxs[idxs_other_params]])
+
+    reset!(agent)
+    @test all(init_params .== agent.problem.p) 
+
+    reset!(env)
+    @test env.current_trial == 1
 end