Fix bounds error check in tests

JuliaReinforcementLearning · May 8, 2024 · 7bda102 · 7bda102
1 parent d99f4df
commit 7bda102
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/episodes.jl b/test/episodes.jl
@@ -116,7 +116,8 @@ using Test
         @test eb.sampleable_inds == [1,1,1,1,1,0,0]
         @test eb[:action][6] == 6
         @test eb[:next_action][5] == 6
-        @test eb[6][:reward] == 0 #6 is not a valid index, the reward there is dummy, filled as zero
+        @test eb[:reward][6] == 0 #6 is not a valid index, the reward there is dummy, filled as zero
+        @test_throws BoundsError eb[6]  #6 is not a valid index, the reward there is dummy, filled as zero
         ep2_len = 0
         for (j,i) = enumerate(8:11)
             ep2_len += 1