Adds extra llamagpt test

mlverse · Apr 8, 2024 · 81461ed · 81461ed
1 parent 15662d1
commit 81461ed
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/tests/testthat/_snaps/backend-llamagpt.md b/tests/testthat/_snaps/backend-llamagpt.md
@@ -39,3 +39,11 @@
       [1] "--threads"   "4"           "--temp"      "0.01"        "--n_predict"
       [6] "1000"        "--model"    
 
+# Output works as expected
+
+    Code
+      ch_llamagpt_output("tests\n> ", stream = TRUE)
+    Output
+      tests
+      [1] "tests\n"
+
diff --git a/tests/testthat/test-backend-llamagpt.R b/tests/testthat/test-backend-llamagpt.R
@@ -63,3 +63,9 @@ test_that("Args output is correct", {
   out <- out[!model_line]
   expect_snapshot(out)
 })
+
+test_that("Output works as expected", {
+  expect_snapshot(
+    ch_llamagpt_output("tests\n> ", stream = TRUE)
+  )
+})