mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-30 12:22:52 +00:00
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
from tqdm import tqdm
|
|
|
|
|
|
def test_silenceable_tqdm_not_disabled_by_default():
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_can_be_silenced_with_0(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "0")
|
|
progress_bar = tqdm(range(1))
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_can_be_silenced_with_false(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "false")
|
|
progress_bar = tqdm(range(1))
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_can_be_silenced_with_False(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "False")
|
|
progress_bar = tqdm(range(1))
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_can_be_silenced_with_FALSE(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "FALSE")
|
|
progress_bar = tqdm(range(1))
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_not_disabled_with_number_above_zero(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "1")
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "10")
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_not_disabled_with_empty_string(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "")
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_not_disabled_with_other_string(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "true")
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "don't print the progress bars please")
|
|
progress_bar = tqdm(range(1))
|
|
assert not progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_can_be_disabled_explicitly():
|
|
progress_bar = tqdm(range(1), disable=True)
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_global_disable_overrides_local_enable(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "0")
|
|
progress_bar = tqdm(range(1), disable=False)
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_global_enable_does_not_overrides_local_disable(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "1")
|
|
progress_bar = tqdm(range(1), disable=True)
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_global_and_local_disable_do_not_clash(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "0")
|
|
progress_bar = tqdm(range(1), disable=True)
|
|
assert progress_bar.disable
|
|
|
|
|
|
def test_silenceable_tqdm_global_and_local_enable_do_not_clash(monkeypatch):
|
|
monkeypatch.setenv("HAYSTACK_PROGRESS_BARS", "1")
|
|
progress_bar = tqdm(range(1), disable=False)
|
|
assert not progress_bar.disable
|