diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
new file mode 100644
index 0000000..539b3da
--- /dev/null
+++ b/.github/workflows/pr.yaml
@@ -0,0 +1,30 @@
+name: PR
+
+on:
+    push:
+        branches: [ "main" ]
+    pull_request:
+        branches: [ "main" ]
+
+jobs:
+    PR:
+        name: Unit test
+        runs-on: ubuntu-latest
+        steps:
+        - uses: actions/checkout@v4
+
+        - name: Install Python
+          uses: actions/setup-python@v5
+          with:
+              python-version: '3.11'
+
+        - name: Setup Python
+          run: |
+              python3 -m pip install --upgrade pip
+              python3 -m pip install pytest
+              python3 -m pip install -r requirements.txt
+
+        - name: Unit test
+          run: |
+              cd models/TestModels
+              python3 -m pytest -v
diff --git a/requirements.txt b/requirements.txt
index d9b9c3f..7ee05d8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ lxml==5.1.0
 pyparsing==3.1.1
 plantuml==0.3.0
 pydantic==2.6.3
+pyyaml==6.0.3