tests.unit package#
Submodules#
tests.unit.test_config module#
- tests.unit.test_config.test_environment_config()#
- tests.unit.test_config.test_lstm_model_config()#
- tests.unit.test_config.test_offline_train_config()#
- tests.unit.test_config.test_offline_train_config_raise_error_no_traj_path()#
- tests.unit.test_config.test_online_train_config()#
- tests.unit.test_config.test_run_config()#
- tests.unit.test_config.test_transformer_model_config()#
tests.unit.test_constants module#
- tests.unit.test_constants.test_twenty_idx_format_func()#
tests.unit.test_environments module#
- tests.unit.test_environments.test_make_env()#
- tests.unit.test_environments.test_make_env_change_view_size()#
- tests.unit.test_environments.test_make_env_flat_one_hot()#
- tests.unit.test_environments.test_make_env_flat_one_hot_view_size_change()#
- tests.unit.test_environments.test_make_env_fully_observed()#
tests.unit.test_ppo_agent module#
- tests.unit.test_ppo_agent.big_transformer_agent(online_config, big_transformer_model_config, environment_config)#
- tests.unit.test_ppo_agent.big_transformer_model_config()#
- tests.unit.test_ppo_agent.environment_config()#
- tests.unit.test_ppo_agent.fc_agent(environment_config)#
- tests.unit.test_ppo_agent.lstm_agent(online_config, lstm_config, environment_config)#
- tests.unit.test_ppo_agent.lstm_config(environment_config)#
- tests.unit.test_ppo_agent.online_config()#
- tests.unit.test_ppo_agent.optimizer()#
- tests.unit.test_ppo_agent.test_fc_agent_init(environment_config)#
- tests.unit.test_ppo_agent.test_fc_agent_layer_init(fc_agent)#
- tests.unit.test_ppo_agent.test_fc_agent_learn(fc_agent, online_config)#
- tests.unit.test_ppo_agent.test_fc_agent_make_optimizer(fc_agent)#
- tests.unit.test_ppo_agent.test_fc_agent_rollout(fc_agent, online_config)#
- tests.unit.test_ppo_agent.test_get_agent_fc_agent(fc_agent, environment_config, online_config)#
- tests.unit.test_ppo_agent.test_get_agent_lstm_agent(lstm_agent, environment_config)#
- tests.unit.test_ppo_agent.test_get_agent_transformer_agent(transformer_agent, transformer_model_config, environment_config, online_config)#
- tests.unit.test_ppo_agent.test_lstm_agent_init(lstm_agent)#
- tests.unit.test_ppo_agent.test_lstm_agent_rollout(lstm_agent, online_config)#
- tests.unit.test_ppo_agent.test_lstm_agent_rollout_learn(lstm_agent, online_config)#
- tests.unit.test_ppo_agent.test_ppo_agent_init()#
- tests.unit.test_ppo_agent.test_ppo_scheduler_step(optimizer)#
- tests.unit.test_ppo_agent.test_transformer_agent_init(transformer_agent)#
- tests.unit.test_ppo_agent.test_transformer_agent_larger_context_init(big_transformer_agent)#
- tests.unit.test_ppo_agent.test_transformer_agent_larger_context_learn(big_transformer_agent, online_config)#
- tests.unit.test_ppo_agent.test_transformer_agent_larger_context_rollout(big_transformer_agent)#
- tests.unit.test_ppo_agent.test_transformer_agent_learn(transformer_agent, online_config)#
- tests.unit.test_ppo_agent.test_transformer_agent_rollout(transformer_agent, online_config)#
- tests.unit.test_ppo_agent.transformer_agent(online_config, transformer_model_config, environment_config)#
- tests.unit.test_ppo_agent.transformer_model_config()#
tests.unit.test_ppo_memory module#
- tests.unit.test_ppo_memory.memory(online_config)#
- tests.unit.test_ppo_memory.online_config()#
- tests.unit.test_ppo_memory.test_add_end_of_episode(memory)#
- tests.unit.test_ppo_memory.test_add_no_ending(online_config)#
- tests.unit.test_ppo_memory.test_compute_advantages(memory)#
- tests.unit.test_ppo_memory.test_get_minibatch_indexes(memory)#
- tests.unit.test_ppo_memory.test_get_minibatch_indexes_recurrence(memory)#
- tests.unit.test_ppo_memory.test_get_minibatches_given_indices_contiguity_of_obs(memory)#
- tests.unit.test_ppo_memory.test_get_minibatches_given_indices_single_env_contiguity_of_obs(memory)#
- tests.unit.test_ppo_memory.test_get_minibatches_recurrance_memory_and_mask(memory)#
- tests.unit.test_ppo_memory.test_get_minibatches_recurrence(memory)#
- tests.unit.test_ppo_memory.test_get_minibatches_standard(memory)#
- tests.unit.test_ppo_memory.test_init(memory)#
- tests.unit.test_ppo_memory.test_minibatch_class()#
tests.unit.test_ppo_utils module#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_MiniGrid()#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_MiniGrid_flat()#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_MiniGrid_full_img_obs()#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_MiniGrid_full_img_partial()#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_MiniGrid_full_obs()#
- tests.unit.test_ppo_utils.test_get_obs_preprocessor_probe1()#
tests.unit.test_streamlit_environment module#
- tests.unit.test_streamlit_environment.decision_transformer_path()#
- tests.unit.test_streamlit_environment.test_get_env_and_dt(decision_transformer_path)#
tests.unit.test_trajectory_model module#
tests.unit.test_utils module#
- tests.unit.test_utils.environment_config()#
- tests.unit.test_utils.online_config()#
- tests.unit.test_utils.run_config()#
- tests.unit.test_utils.test_trajectory_writer_lzma(environment_config, run_config, online_config)#
- tests.unit.test_utils.test_trajectory_writer_numpy(environment_config, run_config, online_config)#
- tests.unit.test_utils.test_trajectory_writer_torch(environment_config, run_config, online_config)#