
    Ng                        d dl Z d dlmZ d dlmZ d dlmZmZ d dlm	Z	 e j
                            de	j        e	j        e	j        e	j        g          d             Zd Zd	 Ze j
                            d
g d          d             Ze j
                            dg d          d             Ze j
                            dddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fddddge	j        fdddg e	j        fg          d             Zd ZdS )    N)example_doc_path)Text)pdf
strategies)PartitionStrategystrategyc                 0    t          j        |            d S )Nr   )r   validate_strategyr
   s    g/var/www/html/ai-engine/env/lib/python3.11/site-packages/test_unstructured/partition/test_strategies.pytest_validate_strategyr   	   s      (333333    c                      t          j        t                    5  t          j        t
          j        d           d d d            d S # 1 swxY w Y   d S NT)r   is_image)pytestraises
ValueErrorr   r   r   FAST r   r   /test_validate_strategy_raises_for_fast_strategyr      s    	z	"	" U U$.?.DtTTTTU U U U U U U U U U U U U U U U U Us   !AAAc                      t          j        t                    5  t          j        d           d d d            d S # 1 swxY w Y   d S )Ntotally_guess_the_text)r   r   r   r   r   r   r   r   .test_validate_strategy_raises_for_bad_strategyr      s    	z	"	" ? ?$%=>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s   <A A )filename	from_fileexpected))layout-parser-paper-fast.pdfTT)copy-protected.pdfTT)loremipsum-flat.pdfTF)r   FT)r   FT)r    FFc                    t          d|            } |r>t          | d          5 }t          j        |          }d d d            n# 1 swxY w Y   nt          j        |           }t	          d |D                       }||u sJ d S )Nzpdf/rb)file)r   c              3   |   K   | ]7}|D ]2}t          |t                    o|j                                        V  38d S )N)
isinstancer   textstrip).0page_elementsels      r   	<genexpr>z/test_is_pdf_text_extractable.<locals>.<genexpr>5   sf           	2t0      r   )r   openr   extractable_elementsany)r   r   r   fextracted_elementspdf_text_extractables         r   test_is_pdf_text_extractabler2   !   s      1x 1 122H I(D!! 	BQ!$!9q!A!A!A	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B !5xHHH  /      8++++++s   AAA)r1   infer_table_structure))TT)FT)TF)FFc                 l    t          j        t          j        | |          }|t          j        k    sJ d S )N)r   r1   r3   )r   determine_pdf_or_image_strategyr   r   )r1   r3   r   s      r   )test_determine_pdf_or_image_fast_strategyr6   >   sD     9"'13  H
 (-------r   )r1   r3   extract_images_in_pdfextract_image_block_typesr   TImageFc                 ^    t          j        t          j        d| |||          }||k    sJ d S )NF)r   r   r1   r3   r7   r8   )r   r5   r   AUTO)r1   r3   r7   r8   r   r   s         r    test_determine_pdf_auto_strategyr<   P   sJ    D 9"'133";  H xr   c                  j    t          j        t          j        d          } | t          j        k    sJ d S r   )r   r5   r   r;   HI_RESr
   s    r   "test_determine_image_auto_strategyr?   }   s?    9"'  H (///////r   )r   test_unstructured.unit_utilsr   unstructured.documents.elementsr   unstructured.partitionr   r   &unstructured.partition.utils.constantsr   markparametrizer;   r   OCR_ONLYr>   r   r   r   r2   r6   r<   r?   r   r   r   <module>rG      s    9 9 9 9 9 9 0 0 0 0 0 0 2 2 2 2 2 2 2 2 D D D D D D " 	 4 4 4
U U U
? ? ?
 )  
 
, ,
 
,$ 5   . . .  
tTG9&7&>?	tT2078	tUWI'8'?@	tUB 1 89	udWI'8'?@	udB 1 89	uegY(9(@A	ueR!2!78	dWI'8'?@	dB 1 89	egY(9(@A	eR!2!9:	tgY(9(@A	tR!2!9:	uwi):)AB	ub"3"<=! 6   7 6 $0 0 0 0 0r   