from unittest.mock import patch

import datasets
from datasets import Dataset
def 				__lowercase	(						)				->Union[str, Any]:

  '''simple docstring'''




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

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

    mock_tqdm.reset_mock()

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


from unittest.mock import patch

import datasets
from datasets import Dataset
def 				__lowercase	(						)				->Union[str, Any]:

  '''simple docstring'''




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

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

    mock_tqdm.reset_mock()

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

