

import unittest

from transformers import load_tool

from .test_tools_common import ToolTesterMixin



class       __snake_case  (    unittest.TestCase     , lowerCAmelCase__       ):

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						load_tool('text-classification')
      self.tool.setup()
      SCREAMING_SNAKE_CASE_									=						load_tool('text-classification'						,		remote=_A)

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						self.tool('That\'s quite cool'						,		['positive', 'negative'])
      self.assertEqual(_A						,		'positive')

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						self.remote_tool('That\'s quite cool'						,		['positive', 'negative'])
      self.assertEqual(_A						,		'positive')

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						self.tool(text='That\'s quite cool'						,		labels=['positive', 'negative'])
      self.assertEqual(_A						,		'positive')


  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						self.remote_tool(text='That\'s quite cool'						,		labels=['positive', 'negative'])
      self.assertEqual(_A						,		'positive')





import pytest

import datasets


# Import fixture modules as plugins
UpperCamelCase__  :      Union[str, Any]			=    ["tests.fixtures.files", "tests.fixtures.hub", "tests.fixtures.fsspec"]


def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		Optional[Any] ,   _SCREAMING_SNAKE_CASE					:		Dict       ):





    """simple docstring"""


    for item in items:
        if any(marker in item.keywords for marker in ['integration', 'unit']       ):
            continue
        item.add_marker(pytest.mark.unit       )


def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		Optional[int]       ):





    """simple docstring"""


    config.addinivalue_line('markers' ,   'torchaudio_latest: mark test to run with torchaudio>=0.12'       )


@pytest.fixture(autouse=_SCREAMING_SNAKE_CASE       )
def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		Any ,   _SCREAMING_SNAKE_CASE					:		str       ):





    """simple docstring"""


    SCREAMING_SNAKE_CASE_									=						tmp_path_factory.getbasetemp() / 'cache'
    SCREAMING_SNAKE_CASE_									=						test_hf_cache_home / 'datasets'
    SCREAMING_SNAKE_CASE_									=						test_hf_cache_home / 'metrics'
    SCREAMING_SNAKE_CASE_									=						test_hf_cache_home / 'modules'
    monkeypatch.setattr('datasets.config.HF_DATASETS_CACHE' ,   str(_SCREAMING_SNAKE_CASE       )       )
    monkeypatch.setattr('datasets.config.HF_METRICS_CACHE' ,   str(_SCREAMING_SNAKE_CASE       )       )
    monkeypatch.setattr('datasets.config.HF_MODULES_CACHE' ,   str(_SCREAMING_SNAKE_CASE       )       )
    SCREAMING_SNAKE_CASE_									=						test_hf_datasets_cache / 'downloads'
    monkeypatch.setattr('datasets.config.DOWNLOADED_DATASETS_PATH' ,   str(_SCREAMING_SNAKE_CASE       )       )
    SCREAMING_SNAKE_CASE_									=						test_hf_datasets_cache / 'downloads' / 'extracted'
    monkeypatch.setattr('datasets.config.EXTRACTED_DATASETS_PATH' ,   str(_SCREAMING_SNAKE_CASE       )       )


@pytest.fixture(autouse=_SCREAMING_SNAKE_CASE ,   scope='session'       )
def 					_UpperCAmelCase       (  ):





    """simple docstring"""


    datasets.disable_progress_bar()


@pytest.fixture(autouse=_SCREAMING_SNAKE_CASE       )
def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		Any       ):





    """simple docstring"""


    monkeypatch.setattr('datasets.config.HF_UPDATE_DOWNLOAD_COUNTS' ,   _SCREAMING_SNAKE_CASE       )





@pytest.fixture
def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		Tuple       ):





    """simple docstring"""


    monkeypatch.setattr('sqlalchemy.util.deprecations.SILENCE_UBER_WARNING' ,   _SCREAMING_SNAKE_CASE       )


