Add cli test

VikParuchuri · Jan 28, 2025 · 49831fb · 49831fb
1 parent 80219ab
commit 49831fb
Show file tree

Hide file tree

Showing 6 changed files with 36 additions and 9 deletions.
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -24,7 +24,7 @@ jobs:
           pip install poetry
           poetry install
 
-      - name: Run detection benchmark test
+      - name: Run benchmark
         run: |
-          poetry run python benchmark.py --max 5 --result_path results --pdftext_only
-          poetry run python scripts/verify_benchmark_scores.py results/results.json
+          poetry run python benchmark/benchmark.py --max 5 --result_path results --pdftext_only
+          poetry run python benchmark/verify_benchmark_scores.py results/results.json
diff --git a/.github/workflows/scripts.yml b/.github/workflows/scripts.yml
@@ -0,0 +1,27 @@
+name: Script test
+
+on: [push]
+
+env:
+  TORCH_DEVICE: "cpu"
+
+jobs:
+  build:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, windows-latest]
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.11
+      - name: Install Python dependencies
+        run: |
+          pip install poetry
+          poetry install
+      - name: Run script
+        run: |
+          poetry run pdftext benchmark/adversarial_short.pdf
diff --git a/benchmark/adversarial_short.pdf b/benchmark/adversarial_short.pdf
diff --git a/benchmark.py → benchmark/benchmark.py b/benchmark.py → benchmark/benchmark.py
diff --git a/scripts/verify_benchmark_scores.py → benchmark/verify_benchmark_scores.py b/scripts/verify_benchmark_scores.py → benchmark/verify_benchmark_scores.py
diff --git a/poetry.lock b/poetry.lock