tests/testthat/test-app.R

In dragonfarm: Fine-Tune Small Language Models with LoRA from R

test_that("chips become templates", {
  expect_equal(dragonfarm:::chips_to_template(c("subject", "body")), "{subject}\\n\\n{body}")
  expect_equal(dragonfarm:::chips_to_template(character()), "")
  expect_equal(dragonfarm:::chips_to_template(NULL), "")
})

test_that("the mapping module produces a mapped dataset", {
  state <- shiny::reactiveValues(dataset = dragon_dataset(toy_df(5)), mapped = NULL)
  shiny::testServer(dragonfarm:::mod_mapping_server, args = list(state = state, nav_to = function(p) NULL), {
    session$setInputs(prompt_tpl = "{subject}\\n\\n{body}", response_tpl = "reply", system_tpl = "")
    m <- mapped()
    expect_s3_class(m, "dragon_dataset")
    expect_equal(m$mapping$prompt, "{subject}\n\n{body}")
    session$flushReact()
    expect_s3_class(state$mapped, "dragon_dataset")

    session$setInputs(prompt_tpl = "{nope}")
    expect_s3_class(mapped(), "mapping_error")
    session$flushReact()
    expect_null(state$mapped)

    session$setInputs(prompt_tpl = "")
    expect_null(mapped())
  })
})

test_that("the train module refuses to launch without inputs", {
  state <- shiny::reactiveValues(dataset = NULL, mapped = NULL, model = NULL, run = NULL)
  shiny::testServer(dragonfarm:::mod_train_server, args = list(state = state, nav_to = function(p) NULL), {
    expect_length(problems(), 3)
    state$dataset <- dragon_dataset(toy_df())
    state$mapped <- dragon_map(state$dataset, "subject", "reply")
    state$model <- list(id = "x/y", trust_remote_code = FALSE)
    expect_length(problems(), 0)
  })
})

test_that("the app object builds", {
  app <- dragon_app(runs_dir = tempfile("runs-"))
  expect_s3_class(app, "shiny.appobj")
})