update documentation

R/pte.R

@@ -514,6 +514,10 @@ pte <- function(yname,
 #'  package will use the empirical bootstrap no matter what the value of this
 #'  parameter.
 #'
+#' @param process_dtt_gt An optional function to customize results when
+#'  the gt-specific function returns the distribution of treated and untreated
+#'  potential outcomes.
+#'
 #' @param ... extra arguments that can be passed to create the correct subsets
 #'  of the data (depending on \code{subset_fun}), to estimate group time
 #'  average treatment effects (depending on \code{attgt_fun}), or to

R/zzz.R

@@ -1 +1 @@
-utils::globalVariables(c("e","post","ciu","cil","G","Y","att","id","name","period","att.e","att.g","group","length.e","length.group","time.period"))
+utils::globalVariables(c("e", "post", "ciu", "cil", "G", "Y", "att", "id", "name", "period", "att.e", "att.g", "group", "length.e", "length.group", "time.period", ".treat"))

tests/testthat/test-did-inference.R

@@ -6,33 +6,36 @@ library(did)
 library(pbapply)
 
 
-skip_inf <- menu(c("run", "skip"), title="would you like to run or skip the inference tests?")
-skip_inf <- skip_inf == 2
+# skip_inf <- menu(c("run", "skip"), title="would you like to run or skip the inference tests?")
+# skip_inf <- skip_inf == 2
 
 test_that("tests for inference", {
-  skip_if(skip_inf)
-  cl <- as.numeric(readline("how many clusters would like to use for the inference tests?  "))
+  #  skip_if(skip_inf)
+  # cl <- as.numeric(readline("how many clusters would like to use for the inference tests?  "))
+  cl <- 1
   mc_sims <- 100
   rejs <- pblapply(1:mc_sims, function(mc) {
     sp <- did::reset.sim()
     data <- did::build_sim_dataset(sp)
-
-    res <- pte(yname="Y",
-               gname="G",
-               tname="period",
-               idname="id",
-               data=data,
-               setup_pte_fun=setup_pte,
-               subset_fun=two_by_two_subset,
-               attgt_fun=did_attgt,
-               xformla=~X)
+
+    res <- pte(
+      yname = "Y",
+      gname = "G",
+      tname = "period",
+      idname = "id",
+      data = data,
+      setup_pte_fun = setup_pte,
+      subset_fun = two_by_two_subset,
+      attgt_fun = did_attgt,
+      xformla = ~X
+    )
     # truth is that att = 1
-    tstat <- (res$overall_att$overall.att-1) / res$overall_att$overall.se
-    rej <- 1*( abs(tstat) > qnorm(.975))
+    tstat <- (res$overall_att$overall.att - 1) / res$overall_att$overall.se
+    rej <- 1 * (abs(tstat) > qnorm(.975))
     rej
-  }, cl=cl)
-  
+  }, cl = cl)
+
   rej_frac <- mean(unlist(rejs))
-  
-  expect_equal(rej_frac, 0.06, tolerance=.05) # make test fail if reject 0
-})
+
+  expect_equal(rej_frac, 0.06, tolerance = .05) # make test fail if reject 0
+})