


'''simple docstring'''




import os
import zipfile

import pytest

from datasets.utils.extract import (
    BzipaExtractor,
    Extractor,
    GzipExtractor,
    LzaExtractor,
    SevenZipExtractor,
    TarExtractor,
    XzExtractor,
    ZipExtractor,
    ZstdExtractor,
)

from .utils import require_lza, require_pyazr, require_zstandard







@pytest.mark.parametrize(
    'compression_format, is_archive'       ,    [
        ('7z', True),
        ('bz2', False),
        ('gzip', False),
        ('lz4', False),
        ('tar', True),
        ('xz', False),
        ('zip', True),
        ('zstd', False),
    ]       ,    )
def _lowerCAmelCase     (      _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    )->     str:
 __UpperCAmelCase					=   {
     '''7z''': (seven_zip_file, SevenZipExtractor),
     '''bz2''': (bza_file, BzipaExtractor),
     '''gzip''': (gz_file, GzipExtractor),
     '''lz4''': (lza_file, LzaExtractor),
     '''tar''': (tar_file, TarExtractor),
     '''xz''': (xz_file, XzExtractor),
     '''zip''': (zip_file, ZipExtractor),
     '''zstd''': (zstd_file, ZstdExtractor),
 }
 __UpperCAmelCase					=   input_paths_and_base_extractors[compression_format]
 if input_path is None:
  __UpperCAmelCase					=   F'for \'{compression_format}\' compression_format, '
  if compression_format == "7z":
   reason += require_pyazr.kwargs["reason"]
  elif compression_format == "lz4":
   reason += require_lza.kwargs["reason"]
  elif compression_format == "zstd":
   reason += require_zstandard.kwargs["reason"]
  pytest.skip(__UpperCamelCase    )
 assert base_extractor.is_extractable(__UpperCamelCase    )
 __UpperCAmelCase					=   tmp_path / ('''extracted''' if is_archive else '''extracted.txt''')
 base_extractor.extract(__UpperCamelCase       ,    __UpperCamelCase    )
 if is_archive:
  assert output_path.is_dir()
  for file_path in output_path.iterdir():
   assert file_path.name == text_file.name
   __UpperCAmelCase					=   file_path.read_text(encoding='utf-8'    )
 else:
  __UpperCAmelCase					=   output_path.read_text(encoding='utf-8'    )
 __UpperCAmelCase					=   text_file.read_text(encoding='utf-8'    )
 assert extracted_file_content == expected_file_content







@pytest.mark.parametrize(
    'compression_format, is_archive'       ,    [
        ('7z', True),
        ('bz2', False),
        ('gzip', False),
        ('lz4', False),
        ('tar', True),
        ('xz', False),
        ('zip', True),
        ('zstd', False),
    ]       ,    )
def _lowerCAmelCase     (      _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    )->     List[str]:
 __UpperCAmelCase					=   {
     '''7z''': seven_zip_file,
     '''bz2''': bza_file,
     '''gzip''': gz_file,
     '''lz4''': lza_file,
     '''tar''': tar_file,
     '''xz''': xz_file,
     '''zip''': zip_file,
     '''zstd''': zstd_file,
 }
 __UpperCAmelCase					=   input_paths[compression_format]
 if input_path is None:
  __UpperCAmelCase					=   F'for \'{compression_format}\' compression_format, '
  if compression_format == "7z":
   reason += require_pyazr.kwargs["reason"]
  elif compression_format == "lz4":
   reason += require_lza.kwargs["reason"]
  elif compression_format == "zstd":
   reason += require_zstandard.kwargs["reason"]
  pytest.skip(__UpperCamelCase    )
 __UpperCAmelCase					=   Extractor.infer_extractor_format(__UpperCamelCase    )
 assert extractor_format is not None
 __UpperCAmelCase					=   tmp_path / ('''extracted''' if is_archive else '''extracted.txt''')
 Extractor.extract(__UpperCamelCase       ,    __UpperCamelCase       ,    __UpperCamelCase    )
 if is_archive:
  assert output_path.is_dir()
  for file_path in output_path.iterdir():
   assert file_path.name == text_file.name
   __UpperCAmelCase					=   file_path.read_text(encoding='utf-8'    )
 else:
  __UpperCAmelCase					=   output_path.read_text(encoding='utf-8'    )
 __UpperCAmelCase					=   text_file.read_text(encoding='utf-8'    )
 assert extracted_file_content == expected_file_content







@pytest.fixture
def _lowerCAmelCase     (      _lowerCAmelCase       ,    _lowerCAmelCase    )->     Tuple:
 import tarfile

 __UpperCAmelCase					=   tmp_path / '''data_dot_dot'''
 directory.mkdir()
 __UpperCAmelCase					=   directory / '''tar_file_with_dot_dot.tar'''
 with tarfile.TarFile(__UpperCamelCase       ,    'w'    ) as f:
  f.add(__UpperCamelCase       ,    arcname=os.path.join('..'       ,    text_file.name    )    )
 return path







@pytest.fixture
def _lowerCAmelCase     (      _lowerCAmelCase    )->     Optional[int]:
 import tarfile

 __UpperCAmelCase					=   tmp_path / '''data_sym_link'''
 directory.mkdir()
 __UpperCAmelCase					=   directory / '''tar_file_with_sym_link.tar'''
 os.symlink('..'       ,    directory / 'subdir'       ,    target_is_directory=__UpperCamelCase    )
 with tarfile.TarFile(__UpperCamelCase       ,    'w'    ) as f:
  f.add(str(directory / 'subdir'    )       ,    arcname='subdir'    )  # str required by os.readlink on Windows and Python < 3.8
 return path







@pytest.mark.parametrize(
    'insecure_tar_file, error_log'       ,    [('tar_file_with_dot_dot', 'illegal path'), ('tar_file_with_sym_link', 'Symlink')]       ,    )
def _lowerCAmelCase     (      _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase    )->     Union[str, Any]:
 __UpperCAmelCase					=   {
     '''tar_file_with_dot_dot''': tar_file_with_dot_dot,
     '''tar_file_with_sym_link''': tar_file_with_sym_link,
 }
 __UpperCAmelCase					=   insecure_tar_files[insecure_tar_file]
 __UpperCAmelCase					=   tmp_path / '''extracted'''
 TarExtractor.extract(__UpperCamelCase       ,    __UpperCamelCase    )
 assert caplog.text
 for record in caplog.records:
  assert record.levelname == "ERROR"
  assert error_log in record.msg







def _lowerCAmelCase     (      _lowerCAmelCase    )->     List[str]:
 __UpperCAmelCase					=   tmpdir / '''not_a_zip_file'''
 # From: https://github.com/python/cpython/pull/5053
 __UpperCAmelCase					=   (
     b'''\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00'''
     b'''\x00\x02\x08\x06\x00\x00\x00\x99\x81\xb6\'\x00\x00\x00\x15I'''
     b'''DATx\x01\x01\n\x00\xf5\xff\x00PK\x05\x06\x00PK\x06\x06\x07'''
     b'''\xac\x01N\xc6|a\r\x00\x00\x00\x00IEND\xaeB`\x82'''
 )
 with not_a_zip_file.open('wb'    ) as f:
  f.write(__UpperCamelCase    )
 assert zipfile.is_zipfile(str(__UpperCamelCase    )    )  # is a false positive for `zipfile`
 assert not ZipExtractor.is_extractable(__UpperCamelCase    )  # but we're right












'''simple docstring'''




from __future__ import annotations







def _lowerCAmelCase     (      _lowerCAmelCase       ,    _lowerCAmelCase       ,    _lowerCAmelCase       ,    )->     tuple:
 if (electron_conc, hole_conc, intrinsic_conc).count(0    ) != 1:
  raise ValueError('You cannot supply more or less than 2 values'    )
 elif electron_conc < 0:
  raise ValueError('Electron concentration cannot be negative in a semiconductor'    )
 elif hole_conc < 0:
  raise ValueError('Hole concentration cannot be negative in a semiconductor'    )
 elif intrinsic_conc < 0:
  raise ValueError(
      'Intrinsic concentration cannot be negative in a semiconductor'    )
 elif electron_conc == 0:
  return (
      "electron_conc",
      intrinsic_conc**2 / hole_conc,
  )
 elif hole_conc == 0:
  return (
      "hole_conc",
      intrinsic_conc**2 / electron_conc,
  )
 elif intrinsic_conc == 0:
  return (
      "intrinsic_conc",
      (electron_conc * hole_conc) ** 0.5,
  )
 else:
  return (-1, -1)


if __name__ == "__main__":
    import doctest

    doctest.testmod()





