fix failing doctest for workflow

cognitedata · Nov 16, 2023 · f7559d7 · f7559d7
1 parent dbaff30
commit f7559d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cognite/client/_api/workflows.py b/cognite/client/_api/workflows.py
@@ -187,8 +187,9 @@ def trigger(
 
             Trigger a workflow execution using a specific set of client credentials (i.e. not your current credentials):
 
+                >>> import os
                 >>> from cognite.client.data_classes import ClientCredentials
-                >>> credentials = ClientCredentials("my-client-id", os.environ["MY_CLIENT_SECRET"]),
+                >>> credentials = ClientCredentials("my-client-id", os.environ["MY_CLIENT_SECRET"])
                 >>> res = c.workflows.executions.trigger("foo", "1", client_credentials=credentials)
         """
         self._warning.warn()

diff --git a/tests/tests_unit/test_docstring_examples.py b/tests/tests_unit/test_docstring_examples.py
@@ -1,4 +1,5 @@
 import doctest
+from collections import defaultdict
 from unittest import TextTestRunner
 from unittest.mock import patch
 
@@ -39,6 +40,7 @@ def run_docstring_tests(module):
 
 
 @patch("cognite.client.CogniteClient", CogniteClientMock)
+@patch("os.environ", defaultdict(lambda: "value"))
 class TestDocstringExamples:
     def test_time_series(self):
         run_docstring_tests(time_series)