






import pytest

import datasets.config
from datasets.utils.info_utils import is_small_dataset
@pytest.mark.parametrize("""dataset_size"""				,				[None, 400 * 2**20, 600 * 2**20]					)
@pytest.mark.parametrize("""input_in_memory_max_size"""				,				["""default""", 0, 100 * 2**20, 900 * 2**20]					)
def 					snake_case_						(   lowerCAmelCase_				:  Optional[Any]				,				lowerCAmelCase_				:  List[Any]				,				lowerCAmelCase_				:  Optional[Any]					):
							if input_in_memory_max_size != "default":
														monkeypatch.setattr(datasets.config				,				"""IN_MEMORY_MAX_SIZE"""				,				lowerCAmelCase_					)
							__lowercase					:							Dict			=   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:
														__lowercase					:							Optional[int]			=   dataset_size < in_memory_max_size
							else:
														__lowercase					:							Any			=   False
							__lowercase					:							List[Any]			=   is_small_dataset(lowerCAmelCase_					)
							assert result == expected







# flake8: noqa
# Lint as: python3
lowerCamelCase       :       Optional[Any]        =				[
    '''VerificationMode''',
    '''Version''',
    '''disable_progress_bar''',
    '''enable_progress_bar''',
    '''is_progress_bar_enabled''',
    '''experimental''',
]

from .info_utils import VerificationMode
from .logging import disable_progress_bar, enable_progress_bar, is_progress_bar_enabled
from .version import Version
from .experimental import experimental