Adds test for env vars

mlverse · Apr 10, 2024 · 9266a4f · 9266a4f
1 parent 86c1681
commit 9266a4f
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/tests/testthat/_snaps/chattr-use.md b/tests/testthat/_snaps/chattr-use.md
@@ -12,3 +12,33 @@
     Output
       [1] "OpenAI - Chat Completions - gpt-4 (gpt4) \n"
 
+# Menu works
+
+    Code
+      chattr_defaults(force = TRUE)
+    Message
+      
+      -- chattr ----------------------------------------------------------------------
+      
+      -- Defaults for: Default --
+      
+      -- Prompt: 
+      * Use the R language, the tidyverse, and tidymodels
+      
+      -- Model 
+      * Provider: LlamaGPT
+      * Path/URL: ~/LlamaGPTJ-chat/build/bin/chat
+      * Model: test/path
+      * Label: GPT4ALL 1.3 (LlamaGPT)
+      
+      -- Model Arguments: 
+      * threads: 4
+      * temp: 0.01
+      * n_predict: 1000
+      
+      -- Context: 
+      Max Data Files: 0
+      Max Data Frames: 0
+      x Chat History
+      x Document contents
+
diff --git a/tests/testthat/test-chattr-use.R b/tests/testthat/test-chattr-use.R
@@ -25,3 +25,14 @@ test_that("Menu works", {
     }
   )
 })
+
+
+test_that("Menu works", {
+  withr::with_envvar(
+    new = c(
+      "CHATTR_USE" = "llamagpt",
+      "CHATTR_MODEL" = "test/path"
+      ),
+    expect_snapshot(chattr_defaults(force = TRUE))
+  )
+})