Skip to main content

gymnasium_rs/
lib.rs

1pub mod core;
2pub mod wrappers;
3pub mod envs;
4
5
6#[cfg(test)]
7mod tests {
8    use crate::core::Env;
9    use super::*;
10
11    #[test]
12    fn it_works() {
13        let mut env = envs::cartpole::CartPole::new();
14        let mut env = wrappers::timelimit::TimeLimitWrapper::new(&mut env, 200);
15
16        let (obs, info) = env.reset_default();
17
18        assert!(obs.len() == 4);
19    }
20}