closes #19

osorensen · osorensen · commit 913f3ec31654 · 2025-04-23T11:24:37.000+02:00
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -20,7 +20,7 @@ LinkingTo:
 Imports: 
     Rcpp
 Depends: 
-    R (>= 2.10)
+    R (>= 4.1.0)
 Suggests: 
     testthat (>= 3.0.0),
     label.switching (>= 1.8)
diff --git a/R/compute_sequentially.R b/R/compute_sequentially.R
@@ -29,7 +29,7 @@
 #'   [precompute_topological_sorts()]. Only used with preference data, and
 #'   defaults to `NULL`.
 #'
-#' @return An object
+#' @return An object of class BayesMallowsSMC2.
 #' @export
 #'
 compute_sequentially <- function(
@@ -79,5 +79,8 @@ compute_sequentially <- function(
 
   ret <- run_smc(input_timeseries, hyperparameters, smc_options,
                  sort_matrices, sort_counts)
+
+  class(ret) <- "BayesMallowsSMC2"
+  ret
 }
 
diff --git a/man/compute_sequentially.Rd b/man/compute_sequentially.Rd
diff --git a/tests/testthat/test-compute_sequentially_complete.R b/tests/testthat/test-compute_sequentially_complete.R
@@ -5,6 +5,7 @@ test_that("compute_sequentially works with complete data", {
     hyperparameters = set_hyperparameters(n_items = 5),
     smc_options = set_smc_options(n_particles = 100, n_particle_filters = 1)
   )
+  expect_s3_class(mod, "BayesMallowsSMC2")
   alpha_hat <- weighted.mean(x = as.numeric(mod$alpha), w = mod$importance_weights)
   expect_gt(alpha_hat, .99)
   expect_lt(alpha_hat, 1.05)

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ test_that("compute_sequentially works with complete data", {`
`5`	`5`	`hyperparameters = set_hyperparameters(n_items = 5),`
`6`	`6`	`smc_options = set_smc_options(n_particles = 100, n_particle_filters = 1)`
`7`	`7`	`)`
	`8`	`+ expect_s3_class(mod, "BayesMallowsSMC2")`
`8`	`9`	`alpha_hat <- weighted.mean(x = as.numeric(mod$alpha), w = mod$importance_weights)`
`9`	`10`	`expect_gt(alpha_hat, .99)`
`10`	`11`	`expect_lt(alpha_hat, 1.05)`