Merge pull request #5 from tjhei/lla-python-mpi-hello-tested

Add Python MPI Hello World example and test it

Author: tjhei

.github/workflows/linux.yml

@@ -129,11 +129,16 @@ jobs:
       with:
         submodules: recursive
     - uses: astral-sh/setup-uv@v6
-    - name: setup
+    - name: Python Setup
       run: |
         uv python install 3.12
         uv venv
         uv sync
+    - name: Python Tests
+      run: |
+        . ./.venv/bin/activate
+        cd cookbooks/landlab/
+        ./test.sh
     - name: compile
       run: |
         mkdir build

cookbooks/landlab/mpi-hello-world/test.py

@@ -0,0 +1,11 @@
+from mpi4py import MPI
+
+comm = MPI.COMM_WORLD
+rank = comm.Get_rank()
+size = comm.Get_size()
+
+sum = comm.allreduce(rank, op=MPI.SUM)
+
+if rank == 0:
+    print(f"Hello World! I am process {rank} of {size}, sum = {sum}")
+

cookbooks/landlab/test.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+set -e
+
+# MPI hello world to see if mpi4py works
+
+cd mpi-hello-world
+mpirun -n 2 python test.py
+cd ..