

'''simple docstring'''






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  (					A__  ,       A__  ,       A__   ):






 """simple docstring"""



 if input_in_memory_max_size != "default":
  monkeypatch.setattr(datasets.config  ,       '''IN_MEMORY_MAX_SIZE'''  ,       A__   )
 __lowercase													=				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													=				dataset_size < in_memory_max_size
 else:
  __lowercase													=				False
 __lowercase													=				is_small_dataset(A__   )
 assert result == expected




'''simple docstring'''






def 	_A  (					):






 """simple docstring"""



 for n in range(1  ,       1000000   ):
  yield n * (n + 1) // 2


def 	_A  (					A__   ):






 """simple docstring"""



 __lowercase													=				1
 __lowercase													=				2
 while i * i <= n:
  __lowercase													=				0
  while n % i == 0:
   n //= i
   multiplicity += 1
  divisors_count *= multiplicity + 1
  i += 1
 if n > 1:
  divisors_count *= 2
 return divisors_count


def 	_A  (					):






 """simple docstring"""



 return next(i for i in triangle_number_generator() if count_divisors(A__   ) > 500   )


if __name__ == "__main__":
 print(solution())

