inst/tinytest/test-getvars.R

In mailund/dfdr: Automatic Differentiation of Simple Functions

library(dfdr)
fl <- dfdr:::init_fct_list()
fct_list <- dfdr:::get_names(fl)
const_fcts <- character()
for(i in fct_list) {
  if(dfdr:::get_keep(fl, fct_list[i])) const_fcts <- c(const_fcts, fct_list[i])
}
  
v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(c[1] <- d[1] + sin(1)))
expect_equal(v$get_ls(), list(quote(c[1])))
expect_equal(v$get_rs(), list(quote(d[1])))
  
v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(c <- d[1] + sin(1)))
expect_equal(v$get_ls(), list(quote(c)))
expect_equal(v$get_rs(), list(quote(d[1])))
  
v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(c[1] <- d + 1))
expect_equal(v$get_ls(), list(quote(c[1])))
expect_equal(v$get_rs(), list(quote(d)))
  
v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(
  if(v < 1) {
    a <- c + r
    h <- 1 + rnorm(3) + j
  }
))
expect_equal(v$get_ls(), list(quote(a), quote(h)))
expect_equal(v$get_rs(), list(quote(c), quote(r), quote(j)))

v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(c <- d + 1))
expect_equal(v$get_ls(), list(quote(c)))
expect_equal(v$get_rs(), list(quote(d)))
  
v <- dfdr:::Vars$new(const_fcts)
ast <- v$find_vars(quote(a <- b))
expect_equal(v$get_ls(), list(quote(a)))
expect_equal(v$get_rs(), list(quote(b)))