Automate token management for local tests

tasks.py

@@ -262,10 +262,6 @@ def tests(context, local=INVOKE_LOCAL):
         context (obj): Used to run specific commands
         local (bool): Define as `True` to execute locally
     """
-    token_openai = os.getenv("OPENAI_TOKEN")
-    if token_openai:
-        del os.environ["OPENAI_TOKEN"]
-
     black(context, local)
     flake8(context, local)
     pylint(context, local)
@@ -275,7 +271,4 @@ def tests(context, local=INVOKE_LOCAL):
     mypy(context, local)
     pytest(context, local)
 
-    if token_openai:
-        os.environ["OPENAI_TOKEN"] = token_openai
-
     print("All tests have passed!")

tests/conftest.py

@@ -0,0 +1,17 @@
+"""Bootstrap Pytest."""
+import os
+
+
+token_openai = os.getenv("OPENAI_TOKEN")
+
+
+def pytest_configure(config):  # pylint: disable=unused-argument
+    """Clean environment for tests."""
+    if token_openai:
+        del os.environ["OPENAI_TOKEN"]
+
+
+def pytest_sessionfinish(session, exitstatus):  # pylint: disable=unused-argument
+    """Recove environment after tests."""
+    if token_openai:
+        os.environ["OPENAI_TOKEN"] = token_openai