Create class for test stats

franzmohr · Jul 8, 2023 · 5bf8df8 · 5bf8df8
1 parent 9b059d4
commit 5bf8df8
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 10 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -55,8 +55,8 @@ Collate:
     'plot.bgvarprd.R'
     'plot.bgvecest.R'
     'plot.ctryvarest.R'
-    'plot.ctryvartest.R'
     'plot.ctryvecest.R'
+    'plot.teststats.bgvarest.R'
     'predict.bgvar.R'
     'summary.bgvarest.R'
     'print.summary.bgvarest.R'

diff --git a/NAMESPACE b/NAMESPACE
@@ -13,8 +13,8 @@ S3method(plot,bgvarirf)
 S3method(plot,bgvarprd)
 S3method(plot,bgvecest)
 S3method(plot,ctryvarest)
-S3method(plot,ctryvartest)
 S3method(plot,ctryvecest)
+S3method(plot,teststats.bgvarest)
 S3method(predict,bgvar)
 S3method(print,summary.bgvarest)
 S3method(print,summary.bgvecest)

diff --git a/R/plot.ctryvartest.R → R/plot.teststats.bgvarest.R b/R/plot.ctryvartest.R → R/plot.teststats.bgvarest.R
@@ -10,7 +10,7 @@
 #' @param ... further graphical parameters.
 #' 
 #' @export
-plot.ctryvartest <- function(x, ctry = NULL, ...) {
+plot.teststats.bgvarest <- function(x, ctry = NULL, ...) {
 
   if (is.null(ctry)) {
     pos <- 1:length(x)

diff --git a/R/select_submodels.R b/R/select_submodels.R
@@ -93,8 +93,8 @@ select_submodels <- function(object, ic = "BIC", select = "order", teststats = N
     criteria <- submodel_test_statistics(object)
   } else {
     # Basic checks
-    if (!"ctryvartest" %in% class(teststats)) {
-      stop("If provided, argument 'teststats' must be of class ctryvartest.")
+    if (!"teststats.bgvarest" %in% class(teststats)) {
+      stop("If provided, argument 'teststats' must be of class teststats.bgvarest")
     }
     criteria <- teststats
   }

diff --git a/R/submodel_test_statistics.R b/R/submodel_test_statistics.R
@@ -257,6 +257,7 @@ submodel_test_statistics <- function(object, ...){
       rownames(result[[current_ctry]][["teststats"]]) <- NULL
       result[[current_ctry]][["loglik"]] <- loglik
       loglik <- list()
+      class(result[[current_ctry]]) <- c("teststats.ctryvarest", "list")
     }
 
     utils::setTxtProgressBar(pb, i / n_models)
@@ -269,7 +270,7 @@ submodel_test_statistics <- function(object, ...){
     return(x)
   })
 
-  class(result) <- c("ctryvartest", "list")
+  class(result) <- c("teststats.bgvarest", "list")
 
   return(result)
 }
diff --git a/man/plot.ctryvartest.Rd → man/plot.teststats.bgvarest.Rd b/man/plot.ctryvartest.Rd → man/plot.teststats.bgvarest.Rd