


import unittest

from transformers.testing_utils import require_bsa
from transformers.utils import is_bsa_available

from ...test_feature_extraction_common import FeatureExtractionSavingTestMixin


if is_bsa_available():
   from transformers import MarkupLMFeatureExtractor






class     UpperCAmelCase__       (					unittest.TestCase		):



    """simple docstring"""






    def __init__(			self					,   A_    )  ->       Any:
         __UpperCamelCase    							=parent






    def _a					(			self    )  ->       Optional[Any]:
         return {}

def 						_UpperCAmelCase				(			):
     __UpperCamelCase    							='<HTML>\n\n    <HEAD>\n    <TITLE>sample document</TITLE>\n    </HEAD>\n\n    <BODY BGCOLOR="FFFFFF">\n    <HR>\n    <a href="http://google.com">Goog</a>\n    <H1>This is one header</H1>\n    <H2>This is a another Header</H2>\n    <P>Travel from\n        <P>\n        <B>SFO to JFK</B>\n        <BR>\n        <B><I>on May 2, 2015 at 2:00 pm. For details go to confirm.com </I></B>\n        <HR>\n        <div style="color:#0000FF">\n            <h3>Traveler <b> name </b> is\n            <p> John Doe </p>\n        </div>'

     __UpperCamelCase    							='\n    <!DOCTYPE html>\n    <html>\n    <body>\n\n    <h1>My First Heading</h1>\n    <p>My first paragraph.</p>\n\n    </body>\n    </html>\n    '

     return [html_string_a, html_string_a]






@require_bsa
class     UpperCAmelCase__       (					A_   ,  unittest.TestCase		):



    """simple docstring"""






    UpperCAmelCase__						:		int      =  MarkupLMFeatureExtractor if is_bsa_available() else None






    def _a					(			self    )  ->       str:
         __UpperCamelCase    							=MarkupLMFeatureExtractionTester(self    )






    @property
    def _a					(			self    )  ->       Union[str, Any]:
         return self.feature_extract_tester.prepare_feat_extract_dict()






    def _a					(			self    )  ->       Tuple:
         # Initialize feature_extractor
         __UpperCamelCase    							=self.feature_extraction_class()

         # Test not batched input
         __UpperCamelCase    							=get_html_strings()[0]
         __UpperCamelCase    							=feature_extractor(A_    )

         # fmt: off
         __UpperCamelCase    							=[['sample document', 'Goog', 'This is one header', 'This is a another Header', 'Travel from', 'SFO to JFK', 'on May 2, 2015 at 2:00 pm. For details go to confirm.com', 'Traveler', 'name', 'is', 'John Doe']]
         __UpperCamelCase    							=[['/html/head/title', '/html/body/a', '/html/body/h1', '/html/body/h2', '/html/body/p', '/html/body/p/p/b[1]', '/html/body/p/p/b[2]/i', '/html/body/p/p/div/h3', '/html/body/p/p/div/h3/b', '/html/body/p/p/div/h3', '/html/body/p/p/div/h3/p']]
         # fmt: on

         self.assertEqual(encoding.nodes					,   A_    )
         self.assertEqual(encoding.xpaths					,   A_    )

         # Test batched
         __UpperCamelCase    							=get_html_strings()
         __UpperCamelCase    							=feature_extractor(A_    )

         # fmt: off
         __UpperCamelCase    							=expected_nodes + [['My First Heading', 'My first paragraph.']]
         __UpperCamelCase    							=expected_xpaths + [['/html/body/h1', '/html/body/p']]

         self.assertEqual(len(encoding.nodes    )					,   2    )
         self.assertEqual(len(encoding.xpaths    )					,   2    )

         self.assertEqual(encoding.nodes					,   A_    )
         self.assertEqual(encoding.xpaths					,   A_    )









from typing import TYPE_CHECKING

from ...utils import _LazyModule


_A		       =						{'tokenization_wav2vec2_phoneme': ['Wav2Vec2PhonemeCTCTokenizer']}


if TYPE_CHECKING:
   from .tokenization_wavaveca_phoneme import WavaVecaPhonemeCTCTokenizer
else:
   import sys

   _A		       =						_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





