python-progressbar/tests/conftest.py at develop · commit-0/python-progressbar · GitHub

History

49 lines (38 loc) · 1.24 KB

Raw

from __future__ import annotations

import logging

import time

import timeit

from datetime import datetime

import freezegun

import pytest

import progressbar

LOG_LEVELS: dict[str, int] = {

'0': logging.ERROR,

'1': logging.WARNING,

'2': logging.INFO,

'3': logging.DEBUG,

}

def pytest_configure(config) -> None:

logging.basicConfig(

level=LOG_LEVELS.get(config.option.verbose, logging.DEBUG),

)

@pytest.fixture(autouse=True)

def small_interval(monkeypatch) -> None:

# Remove the update limit for tests by default

monkeypatch.setattr(

progressbar.ProgressBar,

'_MINIMUM_UPDATE_INTERVAL',

1e-6,

)

monkeypatch.setattr(timeit, 'default_timer', time.time)

@pytest.fixture(autouse=True)

def sleep_faster(monkeypatch):

# The timezone offset in seconds, add 10 seconds to make sure we don't

# accidentally get the wrong hour

offset_seconds = (datetime.now() - datetime.utcnow()).seconds + 10

offset_hours = int(offset_seconds / 3600)

freeze_time = freezegun.freeze_time(tz_offset=offset_hours)

with freeze_time as fake_time:

monkeypatch.setattr('time.sleep', fake_time.tick)

monkeypatch.setattr('timeit.default_timer', time.time)

yield freeze_time

python-progressbar/tests/conftest.py at develop · commit-0/python-progressbar · GitHub

Uh oh!

Expand file tree

Breadcrumbs

conftest.py

Latest commit

History

Breadcrumbs

conftest.py

File metadata and controls