在pytest中支持mpi测试的包
pytest-easyM的Python项目详细描述
欢迎使用pytest easyMPI包!在
这个包的目的是使MPI代码测试与测试类似 尽可能使用常规序列码。希望用户 这个插件可以更专注于编写MPI代码的测试和开销 解决如何将MPI测试集成到其他测试中的时间更少 案例。在
简短示例
下面是一个如何合并串行和并行的示例 在单个测试文件中测试。并行测试使用4个MPI列组运行:
frompytest_easyMPIimportmpi_paralleldeftest_serial():assertTrue@mpi_parallel(4)deftest_parallel():# Import MPI only inside the test that needs it# (to avoid spawning too many MPI communicators)frommpi4pyimportMPIdata=MPI.COMM_WORLD.gather(MPI.COMM_WORLD.Get_rank())ifMPI.COMM_WORLD.Get_rank()==0:assertsum(range(MPI.COMM_WORLD.Get_size()))==sum(data)
然后可以通过调用以下命令运行测试:
^{pr2}$- 项目
标签: