QA: Update tests.

Author: Gadgetoid

tests/conftest.py

@@ -0,0 +1,27 @@
+import sys
+
+import mock
+import pytest
+from tools import SMBusFakeDevice
+
+
+@pytest.fixture()
+def smbus_not_present():
+    sys.modules['smbus2'] = mock.MagicMock()
+    yield sys.modules['smbus2']
+    del sys.modules['smbus2']
+
+
+@pytest.fixture()
+def smbus():
+    sys.modules['smbus2'] = mock.MagicMock()
+    sys.modules['smbus2'].SMBus = SMBusFakeDevice
+    yield sys.modules['smbus2']
+    del sys.modules['smbus2']
+
+
+@pytest.fixture()
+def LSM303D():
+    from lsm303d import LSM303D
+    yield LSM303D
+    del sys.modules['lsm303d']

tests/test_setup.py

@@ -1,22 +1,12 @@
-import sys
-
-import mock
 import pytest
 
 
-def test_setup_not_present():
-    sys.modules['smbus'] = mock.MagicMock()
-    from lsm303d import LSM303D
+def test_setup_not_present(smbus_not_present, LSM303D):
     lsm303d = LSM303D()
     with pytest.raises(RuntimeError):
         lsm303d.setup()
 
 
-def test_setup_mock_present():
-    from tools import SMBusFakeDevice
-    smbus = mock.Mock()
-    smbus.SMBus = SMBusFakeDevice
-    sys.modules['smbus'] = smbus
-    from lsm303d import LSM303D
+def test_setup_mock_present(smbus, LSM303D):
     lsm303d = LSM303D()
     lsm303d.setup()