tests/testthat/test-supervised_model.R

In cregouby/docformer: Fit 'DocFormer' Models for Classification and Regression

test_that("docformer config works with default params", {
  expect_error(config  <-  docformer_config(), NA)
})

test_that("docformer config works with pretrained model params", {
  pretrained_model_name <- "hf-internal-testing/tiny-layoutlm"
  expect_error(
    config  <-  docformer_config(pretrained_model_name=pretrained_model_name),
    NA)
  expect_equal(config$hidden_size, 32L)
  expect_equal(config$vocab_size, 5000L)
})


test_that("docformer config warn if wrong model name", {
  pretrained_model_name <- "inst/tiny-layoutlm"
  expect_warning(
    config  <-  docformer_config(pretrained_model_name=pretrained_model_name),
    "Provided model name cannot be found")
  expect_equal(config$hidden_size, 768L)
  expect_equal(config$vocab_size, 30522L)
})

test_that("docformer config fails if hidden_size is not multiple of num_attention_head", {
  expect_error(config  <-  docformer_config(num_attention_heads = 11L),
               "is not multiple of")
})