教程：测试您的环境

介绍

环境已完成，我们可以测试它以确保其按预期工作。PettingZoo 内置了一个测试套件，可用于测试您的环境。

代码

注意：此代码可以添加到同一文件的底部，无需任何导入，但最佳实践是将测试保存在单独的文件中，并使用模块化导入，如下所示..

出于简便考虑，使用了相对导入，并假设您的自定义环境在同一目录中。如果您的测试位于其他位置（例如，根目录级别的 /test/ 目录），建议使用绝对路径进行导入。

/custom-environment/env/custom_environment.py

from tutorial2_adding_game_logic import CustomEnvironment
from tutorial3_action_masking import CustomActionMaskedEnvironment

from pettingzoo.test import parallel_api_test

if __name__ == "__main__":
    env = CustomEnvironment()
    parallel_api_test(env, num_cycles=1_000_000)

    env = CustomActionMaskedEnvironment()
    parallel_api_test(env, num_cycles=1_000_000)