with_mock_dir("../mocks", { test_that("timegpt_anomaly_detection", { skip_if_no_token() test_data <- nixtlar::electricity response <- timegpt_anomaly_detection(test_data, id_col = "unique_id") expect_s3_class(response, "data.frame") expect_true(all(c("unique_id", "ds", "y", "anomaly", "TimeGPT-lo-99", "TimeGPT", "TimeGPT-hi-99") %in% names(response))) expect_true(all(response$anomaly %in% c(0,1))) expect_true(is.numeric(response$TimeGPT)) expect_true(is.numeric(response$`TimeGPT-lo-99`)) expect_true(is.numeric(response$`TimeGPT-hi-99`)) }) })