Add layer and model tests to CI

Author: Victor-Jung

.github/workflows/CI.yml

@@ -22,3 +22,43 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -e .
+
+  single-layer-tests:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Repo
+        uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+            python-version: "3.11"
+      - name: Build DeepQuant
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e .
+      - name: Run Tests
+        run: |
+          pytest Tests/test_simple_cnn.py
+          pytest Tests/test_simple_mha.py
+          pytest Tests/test_simple_nn.py
+
+  model-tests:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Repo
+        uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+            python-version: "3.11"
+      - name: Build DeepQuant
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e .
+      - name: Run Tests
+        run: |
+          pytest Tests/test_mnist.py