Added GitHub actions for maurorigo

.github/workflows/rigo.yml

@@ -0,0 +1,26 @@
+name: Actions on Rigo
+on: [push]
+jobs:
+  test-docker-image:
+    runs-on: ubuntu-latest
+    container: ghcr.io/maurorigo/jaxubuntu:latest
+
+    steps:
+     - name: Checkout the repo
+       uses: actions/checkout@v3
+     - name: Test for vim
+       run: test -f /bin/vim && echo "Vim is installed."
+     - name: Test for python
+       run: python3 --version
+     - name: Test for numpy
+       run: python3 -c "import numpy" && echo "numpy is installed."
+     - name: Test for jax
+       run: python3 -c "import jax" && echo "jax is installed."
+     - name: Run short test in tests/rigo directory
+       run: |
+          cd tests/rigo
+          python3 test.py
+     - name: Run long test in tests/rigo directory
+       run: |
+          cd tests/rigo
+          python3 MCtest.py

docker/maurorigo/Dockerfile

@@ -0,0 +1,17 @@
+FROM ubuntu:focal
+
+RUN apt-get update \
+    && apt-get install -y vim \
+    && apt-get install -y python3 \
+    && apt-get install -y python3-pip \
+    && pip install numpy \
+    && pip install --upgrade "jax[cpu]"
+RUN echo 'echo "This image runs Ubuntu and has vim, python, pip, numpy and jax installed."' > /usr/local/bin/start.sh
+RUN echo 'echo "For a simple test, run \"python3 test.py\"."' >> /usr/local/bin/start.sh
+RUN chmod +x /usr/local/bin/start.sh
+
+WORKDIR /home
+
+COPY test.py test.py
+
+ENTRYPOINT /usr/local/bin/start.sh && /bin/bash

docker/maurorigo/test.py

@@ -0,0 +1,15 @@
+import numpy as np
+import jax
+jax.config.update('jax_platform_name', 'cpu')
+
+print("Test code using numpy and jax with cpu.")
+
+arr = np.arange(10)
+
+arrnp = np.random.permutation(arr)
+print(f"Permuted with numpy: {arrnp}.")
+
+key = jax.random.PRNGKey(13)
+arrjnp = jax.random.permutation(key, arr)
+assert len(arrjnp) == 10, "Something is wrong with the jax installation."
+print(f"Permuted with jax: {arrjnp}.")