imitation.testing#

Helper methods for unit tests.

May also be useful for users of imitation.

Modules

imitation.testing.expert_trajectories

Test utilities to conveniently generate expert trajectories.

imitation.testing.hypothesis_strategies

Hypothesis strategies for generating sequences of trajectories for testing.

imitation.testing.reward_improvement

Utility functions used to check if rewards improved wrt to previous rewards.

imitation.testing.reward_nets

Utility functions for testing reward nets.