

"""simple docstring"""






import pytest

import datasets.config
from datasets.utils.info_utils import is_small_dataset



@pytest.mark.parametrize("dataset_size"							,      [None, 4_00 * 2**20, 6_00 * 2**20]	)
@pytest.mark.parametrize("input_in_memory_max_size"							,      ["default", 0, 1_00 * 2**20, 9_00 * 2**20]	)
def   lowercase_	(    _lowercase       :		Union[str, Any]							,      _lowercase       :		int							,      _lowercase       :		Union[str, Any]	):




					'''simple docstring'''






					if input_in_memory_max_size != "default":
										monkeypatch.setattr(datasets.config							,      "IN_MEMORY_MAX_SIZE"							,      _lowercase	)
					UpperCAmelCase			:   Optional[int]      =				datasets.config.IN_MEMORY_MAX_SIZE
					if input_in_memory_max_size == "default":
										assert in_memory_max_size == 0
					else:
										assert in_memory_max_size == input_in_memory_max_size
					if dataset_size and in_memory_max_size:
										UpperCAmelCase			:   Tuple      =				dataset_size < in_memory_max_size
					else:
										UpperCAmelCase			:   Optional[int]      =				False
					UpperCAmelCase			:   Union[str, Any]      =				is_small_dataset(_lowercase	)
					assert result == expected





"""simple docstring"""






snake_case_   :   str        =       [
    """DownloadConfig""",
    """DownloadManager""",
    """DownloadMode""",
    """StreamingDownloadManager""",
]

from .download_config import DownloadConfig
from .download_manager import DownloadManager, DownloadMode
from .streaming_download_manager import StreamingDownloadManager


