Running unit tests when issuing a PR (#12)

.github/workflows/format_check.yml

@@ -4,7 +4,7 @@ on:
   pull_request:
   push:
     branches:
-      - master
+      - main
     tags:
       - r[0-9]+.[0-9]+
 

.github/workflows/unit_test.yml

@@ -0,0 +1,26 @@
+name: Unit Tests
+
+on: 
+  workflow_dispatch:
+  pull_request:
+  push:
+    branches:
+      - main
+    tags:
+      - r[0-9]+.[0-9]+
+
+
+jobs:
+  run-shell-script:
+    runs-on: self-hosted
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Run unit test
+      run: |
+        docker pull $UT_IMAGE
+        docker run -v $PWD:$PWD -w $PWD --net host --ipc host --shm-size 80G -t --rm --gpus all $UT_IMAGE bash -c 'pip install -r requirements/requirements-test.txt && bash tests/run_ut.sh'
+      env: 
+        UT_IMAGE: ${{ secrets.UT_IMAGE }}