Update texampleTests

matlab-deep-learning · May 14, 2024 · 996bca1 · 996bca1
1 parent 5341eb4
commit 996bca1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/examples/AnalyzeScientificPapersUsingFunctionCalls.mlx b/examples/AnalyzeScientificPapersUsingFunctionCalls.mlx
diff --git a/tests/texampleTests.m b/tests/texampleTests.m
@@ -5,17 +5,16 @@
 
     methods (TestClassSetup)
         function saveEnvVar(testCase)
-            % Ensures key is not in environment variable for tests
             openAIEnvVar = "OPENAI_KEY";
             key = getenv(openAIEnvVar);
-            unsetenv(openAIEnvVar);
-            testCase.addTeardown(@(x) setenv(openAIEnvVar, x), key);
+            writelines("OPENAI_API_KEY="+key,".env");
+
+            testCase.addTeardown(@() delete(".env"));
+            testCase.addTeardown(@() unsetenv("OPENAI_API_KEY"));
         end
     end
 
     methods(Test)
-        % Test methods
-
         function testAnalyzeScientificPapersUsingFunctionCalls(~)
             AnalyzeScientificPapersUsingFunctionCalls;
         end