Skip to content

Commit

Permalink
add ci for paddleocr test
Browse files Browse the repository at this point in the history
  • Loading branch information
GreatV committed May 7, 2024
1 parent 95e3103 commit 9973b57
Show file tree
Hide file tree
Showing 2 changed files with 72 additions and 0 deletions.
35 changes: 35 additions & 0 deletions .github/workflows/tests.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
name: PaddleOCR PR Tests

on:
push:
pull_request:
branches: ["main", "release/*"]

permissions:
contents: read

jobs:
build:
runs-on: ubuntu-latest

steps:
- uses: actions/checkout@v4
- name: Set up Python 3.10
uses: actions/setup-python@v5
with:
python-version: "3.10"
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install flake8 pytest
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
pip install -e .
- name: Lint with flake8
run: |
# stop the build if there are Python syntax errors or undefined names
flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
- name: Test with pytest
run: |
pytest tests/
37 changes: 37 additions & 0 deletions tests/test_paddleocr_api.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
import pytest
import cv2
from paddleocr import PaddleOCR


@pytest.fixture
def ocr():
return PaddleOCR(lang="en")


@pytest.fixture
def img():
return cv2.imread("doc/imgs/1.jpg")


def test_ocr_with_detection_and_recognition(ocr, img):
result = ocr.ocr(img, det=True, rec=True)
assert result is not None
assert isinstance(result, list)


def test_ocr_with_detection_only(ocr, img):
result = ocr.ocr(img, det=True, rec=False)
assert result is not None
assert isinstance(result, list)


def test_ocr_with_recognition_only(ocr, img):
result = ocr.ocr(img, det=False, rec=True)
assert result is not None
assert isinstance(result, list)


def test_ocr_without_detection_and_recognition(ocr, img):
result = ocr.ocr(img, det=False, rec=False)
assert result is not None
assert isinstance(result, list)

0 comments on commit 9973b57

Please sign in to comment.