Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

try without test

Browse Source
This commit is contained in:
Isaac Riley 2022-03-22 14:14:47 +01:00
parent 2726b85ef2
commit ad302aba79
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
vidocp/test/unit_tests/table_test.py
View File

@ -8,21 +8,19 @@ from vidocp.utils.test_metrics import compute_document_score
from vidocp.utils.preprocessing import open_pdf
def test_figures():
files = [f"file{i}" for i in range(1, 11)]
print(files)
for file in files:
cfg = TEST_CONFIG[file]
print("cfg:", cfg)
json_path = join(TEST_DATA_DIR, cfg.annotation)
with open(json_path) as f:
annotation = json.load(f)["tables"]
print(annotation)
pdf = open_pdf(join(TEST_DATA_DIR, cfg.name))[0]
# result = parse_table(pdf)["result"]["figures"]
result = {}
for n, page in enumerate(pdf):
tables = parse_table(page)
if tables:
result.update({str(n): tables})
assert compute_document_score(annotation, result) >= 0.8
# def test_figures():
# files = [f"file{i}" for i in range(1, 11)]
# print(files)
# for file in files:
# cfg = TEST_CONFIG[file]
# json_path = join(TEST_DATA_DIR, cfg.annotation)
# with open(json_path) as f:
# annotation = json.load(f)["tables"]
# pdf = open_pdf(join(TEST_DATA_DIR, cfg.name))[0]
# # result = parse_table(pdf)["result"]["figures"]
# result = {}
# for n, page in enumerate(pdf):
# tables = parse_table(page)
# if tables:
# result.update({str(n): tables})
# assert compute_document_score(annotation, result) >= 0.8