tests/testthat/test-write_diagram.R

In andreashandel/modeldiagram: Easy Creation of ggplot2 Based Figures of Flow Diagrams

context("test-write_diagram.R")

test_that("write_diagram function works",
          {

            variables <- c("S","I","R")
            varnames <- c("Susceptible","Infected","Recovered")  # optional
            flows <- list(S_flows = c("-b*S*I"),
                          I_flows = c("b*S*I","-g*I"),
                          R_flows = c("g*I"))
            varlocations <-  matrix(data = c("S", "", "R", "", "I", "" ),
                                    nrow = 2, ncol = 3, byrow = TRUE)
            model_list <- list(variables = variables,
                               flows = flows)
            diagram_list <- prepare_diagram(model_list = model_list)

            # generate R code from diagram_list
            write_diagram(diagram_list = diagram_list,
                          filename = "test_code.R",
                          always_overwrite = TRUE)
            testthat::expect_true(fs::file_exists('./test_code.R'))
            fs::file_delete('test_code.R')
          })