Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test_para_pipeline.py 1.87 KB
Newer Older
赵小蒙's avatar
Initial commit  
赵小蒙 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56 
import unittest

"""
Execute the following command to run the test under directory code-clean:

    python -m test.test_para.test_para_pipeline
    
    or
    
    pytest -v -s app/pdf_toolbox/test/test_para/test_para_pipeline.py
    
"""

from test.test_para.test_pdf2text_recogPara_Common import (
    TestIsBboxOverlap,
    TestIsInBbox,
    TestIsBboxOverlap,
    TestIsLineLeftAlignedFromNeighbors,
    TestIsLineRightAlignedFromNeighbors,
)
from test.test_para.test_pdf2text_recogPara_EquationsProcessor import TestCalcOverlapPct
from test.test_para.test_pdf2text_recogPara_BlockInnerParasProcessor import TestIsConsistentLines
from test.test_para.test_pdf2text_recogPara_BlockContinuationProcessor import (
    TestIsAlphabetChar,
    TestIsChineseChar,
    TestIsOtherLetterChar,
)
from test.test_para.test_pdf2text_recogPara_TitleProcessor import TestTitleProcessor


# Test suite
suite = unittest.TestSuite()

# Test cases from test_pdf2text_recogPara_Common
suite.addTest(unittest.makeSuite(TestIsBboxOverlap))
suite.addTest(unittest.makeSuite(TestIsInBbox))
suite.addTest(unittest.makeSuite(TestIsBboxOverlap))
suite.addTest(unittest.makeSuite(TestIsLineLeftAlignedFromNeighbors))
suite.addTest(unittest.makeSuite(TestIsLineRightAlignedFromNeighbors))

# Test cases from test_pdf2text_recogPara_EquationsProcessor
suite.addTest(unittest.makeSuite(TestCalcOverlapPct))

# Test cases from test_pdf2text_recogPara_BlockInnerParasProcessor
suite.addTest(unittest.makeSuite(TestIsConsistentLines))

# Test cases from test_pdf2text_recogPara_BlockContinuationProcessor
suite.addTest(unittest.makeSuite(TestIsAlphabetChar))
suite.addTest(unittest.makeSuite(TestIsChineseChar))
suite.addTest(unittest.makeSuite(TestIsOtherLetterChar))

# Test cases from test_pdf2text_recogPara_TitleProcessor
suite.addTest(unittest.makeSuite(TestTitleProcessor))

# Run test suite
unittest.TextTestRunner(verbosity=2).run(suite)