


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 	__lowerCamelCase						(							lowerCamelCase__		:						Optional[int]  ,							lowerCamelCase__		:						List[str]  ,							lowerCamelCase__		:						Dict      ):





			'''simple docstring'''




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





def 	__lowerCamelCase						(							lowerCamelCase__		:						int      ):





			'''simple docstring'''




			lowerCamelCase    		=  (1 + 24 * n) ** 0.5
			return ((1 + root) / 6) % 1 == 0






def 	__lowerCamelCase						(							lowerCamelCase__		:						int = 5000      ):





			'''simple docstring'''




			lowerCamelCase    		=  [(i * (3 * i - 1)) // 2 for i in range(1  ,							lowerCamelCase__      )]
			for i, pentagonal_i in enumerate(lowerCamelCase__      ):
						for j in range(lowerCamelCase__  ,							len(lowerCamelCase__      )      ):
									lowerCamelCase    		=  pentagonal_nums[j]
									lowerCamelCase    		=  pentagonal_i + pentagonal_j
									lowerCamelCase    		=  pentagonal_j - pentagonal_i
									if is_pentagonal(lowerCamelCase__      ) and is_pentagonal(lowerCamelCase__      ):
												return b

			return -1


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

