Skip to content

Commit f29edc6

Browse files
committed
add some basic tests for env interface
1 parent 0d54f76 commit f29edc6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

test/environments.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@ environments = [
1212
:block
1313
]
1414

15-
for name in environments
15+
@testset "$name" for name in environments
1616
env = get_environment(name)
17-
@test true
17+
@test size(reset(env)) == (env.observation_space.n,)
18+
o, r, d, i = step(env, Dojo.sample(env.input_space))
19+
@test size(o) == (env.observation_space.n,)
1820
end

0 commit comments

Comments
 (0)