X-Git-Url: https://zdv2.bktei.com/gitweb/EVA-2020-02-2.git/blobdiff_plain/87221e4160e484a01d53d66e59db4c57c2904495..c4f5e80eb9b3004836039577369a10cc421db761:/library/tests/conftest.py?ds=sidebyside

diff --git a/library/tests/conftest.py b/library/tests/conftest.py
index b026172..8a5c54c 100644
--- a/library/tests/conftest.py
+++ b/library/tests/conftest.py
@@ -14,6 +14,23 @@ class SMBusFakeDevice(MockSMBus):
         self.regs[0x00:0x01] = 0x0f, 0x00
 
 
+@pytest.fixture(scope='function', autouse=True)
+def cleanup():
+    yield None
+    try:
+        del sys.modules['enviroplus']
+    except KeyError:
+        pass
+    try:
+        del sys.modules['enviroplus.noise']
+    except KeyError:
+        pass
+    try:
+        del sys.modules['enviroplus.gas']
+    except KeyError:
+        pass
+
+
 @pytest.fixture(scope='function', autouse=False)
 def GPIO():
     """Mock RPi.GPIO module."""