test_that("router trains on empirical winners", { log <- data.frame( input = rep(c("simple good text", "complex 123? {}"), each = 2), provider = rep(c("A", "B"), 2), quality = c(.9, .5, .4, .95), stringsAsFactors = FALSE ) r <- train_router(log) expect_s3_class(r, "autogen_router") out <- route_model(r, c("good text", "123? {}")) expect_length(out, 2) })