
    NgU                     j    d dl mZ d dlmZ d dlmZmZ d dlmZm	Z	m
Z
mZ d dlmZ d Zd Zd Zd	 Zd
S )    )defaultdict)Type)elementsontology)ALL_ONTOLOGY_ELEMENT_TYPES$HTML_TAG_TO_DEFAULT_ELEMENT_TYPE_MAP+ONTOLOGY_CLASS_TO_UNSTRUCTURED_ELEMENT_TYPEget_all_subclasses)OntologyElementc                  $   t          t                    } t          D ]/} |            j        D ]}| |                             |           0|                                 D ]/\  }}|t          v sJ t          |         |t          j        gz   v sJ 0d S N)	r   listr   allowed_tagsappenditemsr   r   UncategorizedText)%html_tag_to_possible_ontology_classesontology_classtaghtml_tagpossible_ontology_classess        e/var/www/html/ai-engine/env/lib/python3.11/site-packages/test_unstructured/documents/test_mappings.py0test_if_all_html_tags_have_default_ontology_typer      s    D * 5 N N!>##0 	N 	NC1#6==nMMMM	N 0U/Z/Z/\/\ 
 
++?????3H=AZ&^
 B
 
 
 
 
 

 
    c                  `    t          j                    D ]} t          | t                    sJ d S r   )r   values
issubclassr   )element_types    r   Btest_all_expected_ontology_types_are_subclasses_of_OntologyElementr      s=    <CEE 9 9,8888889 9r   c                      t          j                    D ]=\  } }t          |t          j                  sJ t          | t
          j                  sJ >d S r   )r	   r   r   r   Elementr   r   )ontology_elementunstructured_elements     r   5test_ontology_to_unstructured_mapping_has_valid_typesr$   #   sj     
5	:	<	<F F 	.0@AAAAA*H,DEEEEEEF Fr   c                  T    t          t          j                  D ]} | t          v sJ d S r   )r
   r   r   r	   )r"   s    r   Atest_all_ontology_elements_are_defined_in_mapping_to_unstructuredr&   ,   s@    .x/GHH O O#NNNNNNO Or   N)collectionsr   typingr   unstructured.documentsr   r   unstructured.documents.mappingsr   r   r	   r
   unstructured.documents.ontologyr   r   r   r$   r&    r   r   <module>r-      s    # # # # # #       5 5 5 5 5 5 5 5            < ; ; ; ; ;
 
 
 9 9 9
F F FO O O O Or   