Merge pull request #14 from LiaoU3/codecov

Add code coverage test when PR submitted

.coveragerc

@@ -0,0 +1,3 @@
+[run]
+omit = 
+    tests/*

.github/workflows/code_coverage.yaml

@@ -0,0 +1,46 @@
+name: Code Coverage
+
+on:
+  pull_request:
+    branches:
+      - main
+  workflow_dispatch:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    permissions:
+      pull-requests: write
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: 3.11
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+          pip install coverage
+
+      - name: Run tests with coverage
+        run: |
+          coverage erase
+          coverage run -m unittest discover -s tests
+          result=$(coverage report -m)
+          result_with_backticks="\`\`\`\n$result"
+          echo "$result_with_backticks" > coverage_report.txt
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v2
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+
+      - name: Comment PR
+        uses: thollander/actions-comment-pull-request@v2
+        with:
+          filePath: coverage_report.txt

tests/test_future_market_time.py

@@ -8,6 +8,9 @@ def setUp(self):
         close_dates = ["2024-04-17"]  # Assume 2024/04-17 is close
         fmt.init_close_date_set(close_dates)
 
+    def tearDown(self):
+        fmt.close_date_set = set()
+
     def test__third_wednesday(self):
         third_wednesday = fmt._third_wednesday(2024, 4)
         self.assertEqual(third_wednesday, datetime(2024, 4, 17).date())

tests/test_market_time.py

@@ -8,6 +8,9 @@ def setUp(self):
         close_dates = ["2024-04-04", "2024-04-05"]
         mt.init_close_date_set(close_dates)
 
+    def tearDown(self):
+        mt.close_date_set = set()
+
     def test_init_close_date_set(self):
         self.assertTrue(
             datetime(2024, 4, 4).date() in mt.close_date_set