diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
new file mode 100644
index 0000000..b45af5c
--- /dev/null
+++ b/.github/workflows/test.yaml
@@ -0,0 +1,46 @@
+# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
+# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python
+
+name: Test Mappers
+
+on:
+  workflow_dispatch:
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10", "3.11"]
+
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v3
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install flake8 pytest
+        pip install -r requirements.txt
+    - name: Install package
+      run: |
+        pip install -e .
+    - name: Lint with flake8
+      run: |
+        # stop the build if there are Python syntax errors or undefined names
+        flake8 . \
+            --count \
+            --select=E9,F63,F7,F82 \
+            --show-source \
+            --statistics \
+            --extend-exclude=src/Spectra/attend.py
+        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+    - name: Test with pytest
+      run: |
+        pytest
diff --git a/tests/test_Wormhole.py b/tests/test_Wormhole.py
index 9f41605..5091870 100644
--- a/tests/test_Wormhole.py
+++ b/tests/test_Wormhole.py
@@ -8,8 +8,8 @@
 @pytest.fixture
 def WormholeModel():
     
-    point_cloud_sizes_train = np.random.randint(low = 10, high = 20, size = 64)
-    point_cloud_sizes_test = np.random.randint(low = 10, high = 20, size = 32)
+    point_cloud_sizes_train = np.random.randint(low = 8, high = 16, size = 64)
+    point_cloud_sizes_test = np.random.randint(low = 8, high = 16, size = 32)
     
     pc_train = [np.random.normal(size = [n, 2]) for n in point_cloud_sizes_train]
     pc_test = [np.random.normal(size = [n, 2]) for n in point_cloud_sizes_test]