Merge pull request #682 from MKyhos/680-mod-variables-w-includes-precompile

`mod$variables` works w includes in precompile state (fix #680)

Author: rok-cesnovar

R/model.R

@@ -265,7 +265,11 @@ CmdStanModel <- R6::R6Class(
       invisible(self)
     },
     include_paths = function() {
-      private$include_paths_
+      if (length(self$exe_file()) > 0 && file.exists(self$exe_file())) {
+        return(private$include_paths_)
+      } else {
+        return(private$precompile_include_paths_)
+      }
     },
     code = function() {
       if (length(private$stan_code_) == 0) {
@@ -653,7 +657,11 @@ variables <- function() {
   }
   assert_stan_file_exists(self$stan_file())
   if (is.null(private$variables_) && file.exists(self$stan_file())) {
-    private$variables_ <- model_variables(self$stan_file(), self$include_paths(), allow_undefined = private$using_user_header_)
+    private$variables_ <- model_variables(
+      stan_file = self$stan_file(),
+      include_paths = self$include_paths(),
+      allow_undefined = private$using_user_header_
+    )
   }
   private$variables_
 }

tests/testthat/test-model-variables.R

@@ -108,3 +108,36 @@ test_that("$variables() errors on no stan_file", {
     fixed = TRUE
   )
 })
+
+test_that("$variables() works with #includes, both pre and post compilation.", {
+  
+  data_code <- "
+    data {
+      int N;
+    }
+  "
+  model_code <- "
+    #include data.stan
+    parameters {
+      vector[N] y;
+    }
+    model {
+      y ~ std_normal();
+    }
+  "
+
+  model_file <- write_stan_file(code = model_code)
+  data_file <- write_stan_file(code = data_code, basename = "data.stan")
+
+  mod <- cmdstan_model(
+    stan_file = model_file,
+    include_paths = dirname(data_file),
+    compile = FALSE
+  )
+
+  vars_pre <- mod$variables()
+  mod$compile()
+  vars_post <- mod$variables()
+
+  expect_equal(vars_pre, vars_post)
+})