move epiparameter.print tests to test-print.R

epiverse-trace · Nov 26, 2024 · 39d6fd6 · 39d6fd6
1 parent 4b2e1d9
commit 39d6fd6
Show file tree

Hide file tree

Showing 4 changed files with 58 additions and 58 deletions.
diff --git a/tests/testthat/_snaps/epiparameter.md b/tests/testthat/_snaps/epiparameter.md
diff --git a/tests/testthat/_snaps/print.md b/tests/testthat/_snaps/print.md
@@ -100,3 +100,38 @@
       # i Use `parameter_tbl()` to see a summary table of the parameters.
       # i Explore database online at: https://epiverse-trace.github.io/epiparameter/articles/database.html
 
+# epiparameter.print works as expected
+
+    Code
+      epiparameter(disease = "ebola", epi_name = "incubation", prob_distribution = create_prob_distribution(
+        prob_distribution = "gamma", prob_distribution_params = c(shape = 1, scale = 1)))
+    Message
+      Citation cannot be created as author, year, journal or title is missing
+    Output
+      Disease: ebola
+      Pathogen: NA
+      Epi Parameter: incubation
+      Study: (????). "No citation."
+      Distribution: gamma
+      Parameters:
+        shape: 1.000
+        scale: 1.000
+
+---
+
+    Code
+      epiparameter(disease = "ebola", epi_name = "incubation", prob_distribution = create_prob_distribution(
+        prob_distribution = "gamma", prob_distribution_params = c(shape = 1, scale = 1),
+        discretise = TRUE))
+    Message
+      Citation cannot be created as author, year, journal or title is missing
+    Output
+      Disease: ebola
+      Pathogen: NA
+      Epi Parameter: incubation
+      Study: (????). "No citation."
+      Distribution: discrete gamma
+      Parameters:
+        shape: 1.000
+        scale: 1.000
+
diff --git a/tests/testthat/test-epiparameter.R b/tests/testthat/test-epiparameter.R
@@ -173,29 +173,6 @@ test_that("epiparameter fails as expected", {
   )
 })
 
-test_that("epiparameter.print works as expected", {
-  expect_snapshot(epiparameter(
-    disease = "ebola",
-    epi_name = "incubation",
-    prob_distribution = create_prob_distribution(
-      prob_distribution = "gamma",
-      prob_distribution_params = c(shape = 1, scale = 1)
-    )
-  ))
-})
-
-test_that("epiparameter.print works as expected", {
-  expect_snapshot(epiparameter(
-    disease = "ebola",
-    epi_name = "incubation",
-    prob_distribution = create_prob_distribution(
-      prob_distribution = "gamma",
-      prob_distribution_params = c(shape = 1, scale = 1),
-      discretise = TRUE
-    )
-  ))
-})
-
 test_that("epiparameter.plot does not produce an error", {
   ebola_dist <- suppressMessages(epiparameter(
     disease = "ebola",

diff --git a/tests/testthat/test-print.R b/tests/testthat/test-print.R
@@ -12,3 +12,26 @@ test_that("epiparameter_db print method works as expected for <= 5 <epiparameter
     epiparameter_db(disease = "SARS", epi_name = "offspring distribution")
   )
 })
+
+test_that("epiparameter.print works as expected", {
+  expect_snapshot(epiparameter(
+    disease = "ebola",
+    epi_name = "incubation",
+    prob_distribution = create_prob_distribution(
+      prob_distribution = "gamma",
+      prob_distribution_params = c(shape = 1, scale = 1)
+    )
+  ))
+})
+
+test_that("epiparameter.print works as expected", {
+  expect_snapshot(epiparameter(
+    disease = "ebola",
+    epi_name = "incubation",
+    prob_distribution = create_prob_distribution(
+      prob_distribution = "gamma",
+      prob_distribution_params = c(shape = 1, scale = 1),
+      discretise = TRUE
+    )
+  ))
+})