add tests for plotting functions #58

openjournals/joss-reviews#3888

Author: mcsiple

tests/testthat/test-ggradar.R

@@ -0,0 +1,30 @@
+test_that("check that ggradar produces ggplot object",
+          {
+            plot.data <- data.frame(
+              bycatch = factor(c(
+                "Higher end of bycatch range",
+                "Midpoint of bycatch range",
+                "Lower end of bycatch range"
+              ),
+              ,levels = c(
+                "Higher end of bycatch range",
+                "Midpoint of bycatch range",
+                "Lower end of bycatch range"
+              )),
+              prebuild50 = c(0, 0, 1),
+              prebuild100 = c(0, 0, 1),
+              abundrel10 = c(0, 0, 0.29),
+              abundrel20 = c(0, 0, 0.34),
+              abundrel50 = c(0, 0, 0.53)
+            )
+            
+            p1 <- ggradar(
+              plot.data = plot.data,
+              axis.label.size = 4,
+              grid.label.size = 4,
+              palette.vec = c("#99D594","#E6F598", "#FEE08B") 
+            )
+            
+            expect_true(class(p1)[2]=="ggplot")
+            
+          })

tests/testthat/test-multiplot_proj.R

@@ -0,0 +1,67 @@
+# Test multiplot_proj
+test_that("multiplot_proj returns three-panel figure", {
+  parms <- list(
+    S0 = 0.944, S1plus = 0.99,
+    K1plus = 9000,
+    AgeMat = 18, nages = 20,
+    z = 2.39, lambdaMax = 1.02
+  )
+  InitDepl.vec <- c(0.1, 0.5, 0.9)
+  BycatchCV <- 0.2
+  nyears <- 100
+
+  high.list <- lapply(
+    X = InitDepl.vec,
+    function(x) {
+      projections(
+        NOut = 50,
+        ConstantRateBycatch = list(Rate = 0.3, CV = BycatchCV),
+        InitDepl = x,
+        lh.params = parms,
+        nyears = nyears,
+        obs_CV = 0.2
+      )
+    }
+  )
+
+  med.list <- lapply(
+    X = InitDepl.vec,
+    function(x) {
+      projections(
+        NOut = 50,
+        ConstantRateBycatch = list(Rate = 0.02, CV = BycatchCV),
+        InitDepl = x,
+        lh.params = parms,
+        nyears = nyears,
+        obs_CV = 0.2
+      )
+    }
+  )
+  low.list <- lapply(
+    X = InitDepl.vec,
+    function(x) {
+      projections(
+        NOut = 50,
+        ConstantRateBycatch = list(Rate = 0.001, CV = BycatchCV),
+        InitDepl = x,
+        lh.params = parms,
+        nyears = nyears,
+        obs_CV = 0.2
+      )
+    }
+  )
+  p1 <- multiplot_proj(
+    high.d1 = high.list[[1]], # d1 is the lowest depletion
+    med.d1 = med.list[[1]],
+    low.d1 = low.list[[1]],
+    high.d2 = high.list[[2]],
+    med.d2 = med.list[[2]],
+    low.d2 = low.list[[2]],
+    high.d3 = high.list[[3]],
+    med.d3 = med.list[[3]],
+    low.d3 = low.list[[3]],
+    years.to.plot = nyears
+  )
+
+  expect_equal(3, length(ggplot2::ggplot_build(p1)$layout$layout$PANEL))
+})

tests/testthat/test-plot_proj.R

@@ -0,0 +1,102 @@
+# plot_proj test
+test_that("plot_proj() errors when it tries to plot more years than in projections", {
+  parms <- list(
+    S0 = 0.77, S1plus = 0.88,
+    K1plus = 9000,
+    AgeMat = 3, nages = 10,
+    z = 2.39, lambdaMax = 1.12
+  )
+  initdepl <- 0.5
+  high.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 100,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+  med.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 50,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+
+  low.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 10,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+
+  expect_error(plot_proj(
+    high = high.simple,
+    med = med.simple,
+    low = low.simple,
+    years.plot = 101,
+    ylims = c(0, parms$K1plus), InitDepl = initdepl,
+    K1plus = parms$K1plus
+  ))
+})
+
+test_that("plot_proj() errors when it tries to plot more years than in projections", {
+  parms <- list(
+    S0 = 0.77, S1plus = 0.88,
+    K1plus = 9000,
+    AgeMat = 3, nages = 10,
+    z = 2.39, lambdaMax = 1.12
+  )
+  initdepl <- 0.5
+  high.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 100,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+  med.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 50,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+
+  low.simple <- projections(
+    NOut = 50,
+    ConstantBycatch = list(
+      Catch = 10,
+      CV = 0.3
+    ),
+    InitDepl = initdepl,
+    lh.params = parms,
+    nyears = 100
+  )
+
+  p1 <- plot_proj(
+    high = high.simple,
+    med = med.simple,
+    low = low.simple,
+    years.plot = 100,
+    ylims = c(0, parms$K1plus), InitDepl = initdepl,
+    K1plus = parms$K1plus
+  )
+
+  expect_true(ggplot2::is.ggplot(p1))
+})