add unit test for trainer sdk

.github/workflows/test-unit.yaml

@@ -0,0 +1,32 @@
+name: test-unit
+
+on:
+  - pull_request
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+      - uses: pre-commit/action@v3.0.1
+
+  coverage:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Run Tests
+        run: |
+          make test-unit xml=true
+
+      - name: Upload coverage to Coveralls
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          coveralls --verbose || echo "Coveralls report failed"

.gitignore

@@ -9,3 +9,5 @@ __debug_bin
 # Python cache files
 __pycache__/
 *.egg-info/
+
+.coverage

Makefile

@@ -0,0 +1,11 @@
+# make test-unit will produce html coverage by default. Run with `make test-unit report=xml` to produce xml report.
+.PHONY: test-unit
+test-unit:
+	pip install "./python[test]"
+	coverage run --source=kubeflow.trainer.api.trainer_client,kubeflow.trainer.utils.utils -m pytest ./python/kubeflow/trainer/api/trainer_client_test.py
+	coverage report -m kubeflow/trainer/api/trainer_client.py kubeflow/trainer/utils/utils.py
+ifeq ($(report),xml)
+	coverage xml
+else
+	coverage html
+endif