Function test_progressbar

mne/utils/tests/test_progressbar.py:15–53

Test progressbar class.

(monkeypatch)

def test_progressbar(monkeypatch):
    """Test progressbar class."""
    a = np.arange(10)
    pbar = ProgressBar(a)
    assert a is pbar.iterable
    assert pbar.max_value == 10

    pbar = ProgressBar(10)
    assert pbar.max_value == 10
    assert pbar.iterable is None

    # Make sure that non-iterable input raises an error
    def iter_func(a):
        for ii in a:
            pass

    with pytest.raises(TypeError, match="not iterable"):
        iter_func(pbar)

    # Make sure different progress bars can be used
    monkeypatch.setenv("MNE_TQDM", "tqdm")
    with catch_logging("debug") as log, ProgressBar(np.arange(3)) as pbar:
        for p in pbar:
            pass
    log = log.getvalue()
    assert "Using ProgressBar with tqdm\n" in log
    monkeypatch.setenv("MNE_TQDM", "broken")
    with pytest.raises(ValueError, match="Invalid value for the"):
        ProgressBar(np.arange(3))
    monkeypatch.setenv("MNE_TQDM", "tqdm.broken")
    with pytest.raises(ValueError, match="Unknown tqdm"):
        ProgressBar(np.arange(3))
    # off
    monkeypatch.setenv("MNE_TQDM", "off")
    with catch_logging("debug") as log, ProgressBar(np.arange(3)) as pbar:
        for p in pbar:
            pass
    log = log.getvalue()
    assert "Using ProgressBar with off\n" == log


def _identity(x):