:QgL8ddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ejjdeded gd Zd Zd Zd ZdZejjdgddZdZdZdZdZdZdZy))mockN)example_doc_path)TableAlignment)TableEvalProcessor)eval_table_transformer_for_fileimage_or_pdf_to_dataframefilename$img/table-multi-row-column-cells.pngz$pdf/table-multi-row-column-cells.pdfc>t|}gd|jvsJy)N)Blind514z 34.5%, n=1z 1199 sec, n=1)rvalues)r dfs k/var/www/html/answerous/venv/lib/python3.12/site-packages/test_unstructured/metrics/test_table_structure.pytest_image_or_pdf_to_dataframers  #8 ,B Bbii OO Oc`ttdtd}d|cxkrdksJJy)Nr z'table-multi-row-column-cells-actual.csvg?)rr)scores r$test_eval_table_transformer_for_filers5 +?@BC E ????rc ddddidg}dddddddd d ddddd d d ddddd d ddddddd gdg}t||}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)NTable text_as_htmlz
r1c1r1c2
r2c1r2c2
typemetadata$ee862c7a-d27e-4484-92de-4faa42a63f3brrr1c1idxywhcontent$6237ac7b-bfc8-40d2-92f2-d138277205e2r2c1$9d0933a9-5984-4cad-80d9-6752bf9bc4dfr1c2$1152d043-5ead-4ab8-8b88-888d48831ac2r2c2rtext?r process_file total_tablestable_level_accelement_row_level_index_accelement_col_level_index_accelement_row_level_content_accelement_col_level_content_acc prediction ground_truth te_processorresults r test_table_eval_processor_simpler>$s4!R JA% A% A% A% 3!$ &LP&j,?L  & & (F   ! ## #  ! !S (( (  - - 44 4  - - 44 4  / /3 66 6  / /3 66 6rc ddddddddddddddddddddddddd dgid g}dd dddddd d dddddd dddddd d ddddddd gdg}t||d}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)Nrtable_as_cellsrr-)r#r$r%r&r'rr r)r+rrr!r(r*r,r.cells source_typer0r1r9s r4test_table_eval_processor_simple_when_input_as_cellsrDas #) #) #) #) -#" $JPA% A% A% A% 3!$ &LP&j,GTL  & & (F   ! ## #  ! !S (( (  - - 44 4  - - 44 4  / /3 66 6  / /3 66 6rcddgidg}dgdg}t||d}tjt5|j dddy#1swYyxYw)Nrr@rr. wrong_typerB)rpytestraises ValueErrorr2)r:r;r<s r0test_table_eval_processor_when_wrong_source_typerJsl)2. J L&j,LYL z "$!!#$$$s AAr)a1
r1c1 r1c2
r2c1 r2c2
r3c1 r3c2
a
r1c1 r1c2
r2c1 r2c2
r3c1 r3c2
a
r1c1 r1c2
r2c1 r2c2
r3c1 r3c2
c dd|idg}ddddddddd ddddd dd ddddd dd ddddddddddddddddddddgdg}t||}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)Nrrrrrrr r!r(r)r*r+r,r-z$364f4a17-2979-4506-ae77-e8adf8e3f554r3c1z$30f87503-ac1f-4db1-b924-b316af585702r3c2r.r0r1)rr:r;r<r=s r7test_table_eval_processor_various_table_html_structuresrOs_|#0NOPJA% A% A% A% A% A% S14 6Lp&j,?L  & & (F   ! ## #  ! !S (( (  - - 44 4  - - 44 4  / /3 66 6  / /3 66 6rc ddddidg}dddddddd d ddddd d d ddddd d ddddddd gdg}t||}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)Nrrz
11 12
21 22
rrrr11r!r(21r*12r,22r.r0r1r9s r1test_table_eval_processor_non_str_values_in_tablerUPs3!  J2A# A# A# A# 3!$ &LP&j,?L  & & (F   ! ## #  ! !S (( (  - - 44 4  - - 44 4  / /3 66 6  / /3 66 6rcdddidg}ddddddd d d ddddd d d dddddd ddddddd ddddddd ddddddd ddddddd ddddddd ddddddd ddddddd ddddddd g d g}t||}|j}|jdk(sJ|jd!k(sJ|jd!k(sJ|j d!k(sJ|j d!k(sJ|jd!k(sJy)"Nrra
r1c1 r1c2 r1c3
r2c2 r2c3 r2c4
r3c1 r3c2 r3c3
r4c1 r4c2
rz$f399ef57-5b88-4509-8971-9cb63246866errrLr r!z$2dfdec2f-e8f3-4be7-a6ac-8ff21c4e8556rMz$9c771c58-88c7-49d8-9c12-85d0e44b920er4c1z$5bd6f3f0-34c5-495b-8a28-c4ac96989ef8r+z$7b8e6bc2-a310-4dd6-997c-313f951e7f96r-z$1c152ad4-12fa-4a7b-90de-a992aa6410a4rNz$55063f64-0003-4217-b6ca-aff5914793ffr4c2z$22852e86-0e22-4d32-b63a-9ba7dd4118a2r1c3z$eae013c5-5597-4a8b-9771-82e28c5c5cbar2c3z$0dea3a42-8523-4d6e-9e70-d65cc2314678r3c3z$60093e2c-d3e2-4146-92b5-97a2fc16c061r2c4r.r0r1r9s r&test_table_eval_processor_merged_cellsr^s! !JJA% A% A% A% A% A% A% A% A% A% A% cY\ ^L@&j,?L  & & (F   ! ## #  ! !S (( (  - - 44 4  - - 44 4  / /3 66 6  / /3 66 6rc dddidg}dddddddd d ddddd d d ddddd d ddddddd gdg}tjjtd5}dg|_t ||}|j }dddjdk(sJ|jdk(sJ|jdk(sJ|jdk(sJ|jdk(sJ|jdk(sJy#1swYpxYw)Nrr*
Some cell
rrrrrQr!r(rRr*rSr,rTr.get_table_level_alignment) rpatchobjectr return_valuerr2r3r4r5r6r7r8)r:r;align_fnr<r=s r1test_table_eval_processor_when_no_match_with_predrg(sg')YZ JA# A# A# A# 3!$ &LP   >+F G-8!#)*lC **,-   ! ## #  ! !Q && &  - - 22 2  - - 22 2  / /1 44 4  / /1 44 4--s %C''C0cig}ig}t||}|j}|jdk(sJ|jdk(sJt j |j sJt j |jsJt j |jsJt j |jsJy)Nrr) rr2r3r4npisnanr5r6r7r8r9s r(test_table_eval_processor_when_no_tablesrkesJ4L%j,?L  & & (F   ! ## #  ! !Q && & 88F66 77 7 88F66 77 7 88F88 99 9 88F88 99 9rc Xg}dddddddddddddddd ddddd dd ddddd dgd g}t||}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)NrrrrrQr!r(rRr*rSr,rTr.r1r9s r&test_table_eval_processor_when_only_gtrmtsJA# A# A# A# 3!$ &LP&j,?L  & & (F   ! ## #  ! !Q && &  - - 22 2  - - 22 2  / /1 44 4  / /1 44 4rcdddidg}ig}t||}|j}|jdk(sJ|jdk(sJ|jdk(sJ|j dk(sJ|j dk(sJ|jdk(sJy)Nrrr`rrr1r9s r(test_table_eval_processor_when_only_predros')YZ J4L%j,?L  & & (F   ! ## #  ! !Q && &  - - 22 2  - - 22 2  / /1 44 4  / /1 44 4r)unittestrnumpyrirGtest_unstructured.unit_utilsr*unstructured.metrics.table.table_alignmentr%unstructured.metrics.table.table_evalr$unstructured.metrics.table_structurerrmark parametrizerrr>rDrJrOrUr^rgrkrmrorrrys 9ED ?@?@PP :7zV7r$(9<zB7{<zB7JG7TK7\:5z :35l5r