Test tweaks and linting

[EVA-2020-02-2.git] / library / tests / conftest.py

"""Test configuration.\r
These allow the mocking of various Python modules\r
that might otherwise have runtime side-effects.\r
"""\r
import sys\r
import mock\r
import pytest\r
from i2cdevice import MockSMBus\r
\r
\r
class SMBusFakeDevice(MockSMBus):\r
    def __init__(self, i2c_bus):\r
        MockSMBus.__init__(self, i2c_bus)\r
        self.regs[0x00:0x01] = 0x0f, 0x00\r
\r
\r
@pytest.fixture(scope='function', autouse=False)\r
def GPIO():\r
    """Mock RPi.GPIO module."""\r
    GPIO = mock.MagicMock()\r
    # Fudge for Python < 37 (possibly earlier)\r
    sys.modules['RPi'] = mock.Mock()\r
    sys.modules['RPi'].GPIO = GPIO\r
    sys.modules['RPi.GPIO'] = GPIO\r
    yield GPIO\r
    del sys.modules['RPi']\r
    del sys.modules['RPi.GPIO']\r
\r
\r
@pytest.fixture(scope='function', autouse=False)\r
def spidev():\r
    """Mock spidev module."""\r
    spidev = mock.MagicMock()\r
    sys.modules['spidev'] = spidev\r
    yield spidev\r
    del sys.modules['spidev']\r
\r
\r
@pytest.fixture(scope='function', autouse=False)\r
def smbus():\r
    """Mock smbus module."""\r
    smbus = mock.MagicMock()\r
    smbus.SMBus = SMBusFakeDevice\r
    sys.modules['smbus'] = smbus\r
    yield smbus\r
    del sys.modules['smbus']\r
\r
\r
@pytest.fixture(scope='function', autouse=False)\r
def atexit():\r
    """Mock atexit module."""\r
    atexit = mock.MagicMock()\r
    sys.modules['atexit'] = atexit\r
    yield atexit\r
    del sys.modules['atexit']\r
\r