refactor assign_round_number

WorldHealthOrganization · Feb 6, 2025 · 576490a · 576490a
1 parent 3cdd61f
commit 576490a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/testthat/test-assign_round_number.R b/tests/testthat/test-assign_round_number.R
@@ -5,10 +5,10 @@ testthat::test_that("assign_round_number works with basic functionality", {
 
   result <- assign_round_number(df, date_col = "today", threshold = 7)
 
-  expected_groups <- c("April 2024", "April 2024", "April 2024", "June 2024")
+  expected_groups <- c("Apr 2024", "Apr 2024", "Apr 2024", "Jun 2024")
 
   testthat::expect_equal(levels(result$round_group), 
-                         c("April 2024", "May 2024","June 2024"))
+                         c("Apr 2024", "May 2024","Jun 2024"))
 })
 
 
@@ -23,11 +23,11 @@ testthat::test_that("assign_round_number works with basic functionality", {
   result <- assign_round_number(df, date_col = "today", threshold = 7,
                                 additional_grouping = "states")
 
-  expected_groups <- c("April 2024", "April 2024", "April 2024", 
-                       "June 2024", "April 2024", "April 2024", "April 2024")
+  expected_groups <- c("Apr 2024", "Apr 2024", "Apr 2024", 
+                       "Jun 2024", "Apr 2024", "Apr 2024", "Apr 2024")
 
   testthat::expect_equal(levels(result$round_group), 
-                         c("April 2024", "May 2024","June 2024"))
+                         c("Apr 2024", "May 2024","Jun 2024"))
 })