





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      _A       (		SCREAMING_SNAKE_CASE     :   Union[str, Any]       ,       SCREAMING_SNAKE_CASE     :   Optional[Any]       ,       SCREAMING_SNAKE_CASE     :   Any						):





  """simple docstring"""



  if input_in_memory_max_size != "default":
    monkeypatch.setattr(datasets.config       ,       "IN_MEMORY_MAX_SIZE"       ,       SCREAMING_SNAKE_CASE						)
  a__	:					Union[str, Any]										=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:
    a__	:					List[str]										=dataset_size < in_memory_max_size
  else:
    a__	:					Any										=False
  a__	:					List[str]										=is_small_dataset(SCREAMING_SNAKE_CASE						)
  assert result == expected








'''simple docstring'''







def 				SCREAMING_SNAKE_CASE__	(   __A = 100	)       ->	int:
	_snake_case   						=		n * (n + 1) * (2 * n + 1) / 6
	_snake_case   						=		(n * (n + 1) / 2) ** 2
	return int(square_of_sum - sum_of_squares	)


if __name__ == "__main__":
					print(F'''{solution() = }''')

