tests/testthat/test-models.R

In rollama: Communicate with 'Ollama' to Run Large Language Models Locally

options("rollama_verbose" = FALSE)

test_that("pull model", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_equal(nrow(pull_model()), 1L)
})

test_that("show model", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_equal(nrow(show_model()), 1L)
  expect_equal(ncol(list_models()), 11L)
  expect_s3_class(list_models(), "tbl_df")
})

test_that("create model", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_equal(nrow(create_model(
    model = "mario",
    modelfile = "FROM llama3.1\nSYSTEM You are mario from Super Mario Bros."
  )), 1L)
  # also test modelfile
  expect_equal(nrow(create_model(
    model = "mario",
    modelfile = system.file("extdata", "modelfile.txt", package = "rollama")
  )), 1L)
})

test_that("copy model", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_message(copy_model("mario"),
                 "model.mario.copied.to.mario-copy")
})

test_that("delete model", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_message(delete_model("mario"),
                 "model.mario.removed")
  expect_message(delete_model("mario-copy"),
                 "model.mario-copy.removed")
})

test_that("model missing", {
  skip_if_not(ping_ollama(silent = TRUE))
  expect_error(check_model_installed("NOMODEL"),
               "Model NOMODEL not installed.")
})