





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       A__     ( unittest.TestCase    ):





      def __init__(				self ,							A_     ):
       '''simple docstring'''




       UpperCamelCase			:      List[str] 			=     parent





      def  __UpperCamelCase(				self     ):
       '''simple docstring'''




       return {}



def 	A_       (  )		->				Tuple:
 UpperCamelCase			:      Optional[int] 			=     "<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			:      List[Any] 			=     "\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       A__     ( __lowerCamelCase ,     unittest.TestCase    ):
      _UpperCAmelCase			:int							=     MarkupLMFeatureExtractor if is_bsa_available() else None





      def  __UpperCamelCase(				self     ):
       '''simple docstring'''




       UpperCamelCase			:      List[Any] 			=     MarkupLMFeatureExtractionTester(self     )





      @property
      def  __UpperCamelCase(				self     ):
       '''simple docstring'''




       return self.feature_extract_tester.prepare_feat_extract_dict()





      def  __UpperCamelCase(				self     ):
       '''simple docstring'''




       UpperCamelCase			:      Tuple 			=     self.feature_extraction_class()

       # Test not batched input
       UpperCamelCase			:      List[Any] 			=     get_html_strings()[0]
       UpperCamelCase			:      Dict 			=     feature_extractor(SCREAMING_SNAKE_CASE_     )

       # fmt: off
       UpperCamelCase			:      int 			=     [["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			:      List[str] 			=     [["/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 ,							SCREAMING_SNAKE_CASE_     )
       self.assertEqual(encoding.xpaths ,							SCREAMING_SNAKE_CASE_     )

       # Test batched
       UpperCamelCase			:      str 			=     get_html_strings()
       UpperCamelCase			:      Dict 			=     feature_extractor(SCREAMING_SNAKE_CASE_     )

       # fmt: off
       UpperCamelCase			:      List[str] 			=     expected_nodes + [["My First Heading", "My first paragraph."]]
       UpperCamelCase			:      Union[str, Any] 			=     expected_xpaths + [["/html/body/h1", "/html/body/p"]]

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

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








from ..utils import DummyObject, requires_backends



class       A__     ( metaclass=__snake_case    ):
      _UpperCAmelCase			:Tuple							=     ['note_seq']





      def __init__(				self ,							*A_ ,							**A_     ):
       '''simple docstring'''




       requires_backends(self ,							["note_seq"]     )





      @classmethod
      def  __UpperCamelCase(				cls ,							*A_ ,							**A_     ):
       '''simple docstring'''




       requires_backends(cls ,							["note_seq"]     )





      @classmethod
      def  __UpperCamelCase(				cls ,							*A_ ,							**A_     ):
       '''simple docstring'''




       requires_backends(cls ,							["note_seq"]     )
