
from unittest.mock import patch

import datasets
from datasets import Dataset



def 		UpperCamelCase    (					)      -> List[Any]:






						'''simple docstring'''

						_lowercase					:  Any    			=      Dataset.from_dict({"col_1": [3, 2, 0, 1]}     )

						with patch("tqdm.auto.tqdm"     ) as mock_tqdm:
												datasets.disable_progress_bar()
												dset.map(lambda _UpperCAmelCase     : {"col_2": x["col_1"] + 1}     )
												mock_tqdm.assert_not_called()

												mock_tqdm.reset_mock()

												datasets.enable_progress_bar()
												dset.map(lambda _UpperCAmelCase     : {"col_2": x["col_1"] + 1}     )
												mock_tqdm.assert_called()






def 		UpperCamelCase    (					_UpperCAmelCase     :      int     )      -> bool:






						'''simple docstring'''

						if not isinstance(_UpperCAmelCase   , _UpperCAmelCase     ):
												_lowercase					:  Optional[int]    			=      f"""Input value of [number={number}] must be an integer"""
												raise TypeError(_UpperCAmelCase     )
						if number < 0 or number % 2 == 1:
												return False
						_lowercase					:  List[str]    			=      int(number**0.5     )
						return number == number_sqrt * (number_sqrt + 1)


if __name__ == "__main__":
							import doctest

							doctest.testmod()




