somet tests

potatoqualitee · Jun 29, 2024 · bde406c · bde406c
1 parent 4b293ce
commit bde406c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 21 deletions.
diff --git a/finetuna.psm1 b/finetuna.psm1
@@ -25,6 +25,7 @@ function Import-ModuleFile {
     else { $ExecutionContext.InvokeCommand.InvokeScript($false, ([scriptblock]::Create([io.file]::ReadAllText($Path))), $null, $null) }
 }
 
+# import internal functions from PSOpenAI bc I have changes
 foreach ($file in (Get-ChildItem -Path $privatedir -Filter *.ps1)) {
     . Import-ModuleFile -Path $file.FullName
 }
@@ -39,6 +40,7 @@ foreach ($function in (Get-ChildItem "$ModuleRoot\public" -Filter "*.ps1" -Recur
     . Import-ModuleFile -Path $function.FullName
 }
 
+# i need these psopenai public things to use my internal functions
 foreach ($file in (Get-ChildItem -Path $publicdir -Filter *.ps1)) {
     . Import-ModuleFile -Path $file.FullName
 }

diff --git a/tests/finetuna.Tests.ps1 b/tests/finetuna.Tests.ps1
@@ -1,5 +1,5 @@
 
-BeforeAll {
+BeforeEach {
     if (-not $env:OPENAI_API_KEY) {
         throw "OPENAI_API_KEY environment variable is not set."
     }
@@ -14,6 +14,14 @@ BeforeAll {
 }
 
 Describe "finetuna Module Tests" {
+    Context "Clear-TuneProvider" {
+        It "Should clear the provider configuration" {
+            Clear-TuneProvider
+            $provider = Get-TuneProvider
+            $provider.ApiKey | Should -BeNullOrEmpty
+            $provider.ApiType | Should -BeNullOrEmpty
+        }
+    }
     Context "Set-TuneProvider" {
         It "Should set the API key and configuration" {
             $splat = @{
@@ -25,17 +33,6 @@ Describe "finetuna Module Tests" {
             $provider.ApiType | Should -Be 'openai'
             (Get-OpenAIContext).ApiKey | Should -Not -BeNullOrEmpty
         }
-
-        $context = Get-OpenAIContext
-        $global:PSDefaultParameterValues['Get-OpenAIAPIParameter:Parameters'] = @{
-            ApiKey        = $context.ApiKey
-            AuthType      = $context.AuthType
-            Organization  = $context.Organization
-            ApiBase       = $context.ApiBase
-            ApiVersion    = $context.ApiVersion
-            TimeoutSec    = $context.TimeoutSec
-            MaxRetryCount = $context.MaxRetryCount
-        }
     }
 
     Context "Get-TuneProvider" {
@@ -116,15 +113,6 @@ Describe "finetuna Module Tests" {
         }
     }
 
-    Context "Clear-TuneProvider" {
-        It "Should clear the provider configuration" {
-            Clear-TuneProvider
-            $provider = Get-TuneProvider
-            $provider.ApiKey | Should -BeNullOrEmpty
-            $provider.ApiType | Should -BeNullOrEmpty
-        }
-    }
-
     Context "Get-Embedding" {
         It "Should generate an embedding for the given text" {
             $result = Get-Embedding -Text "Hello, world!"