Test simple example for tundraContainer train.

syberia · Aug 2, 2015 · 4cf3a25 · 4cf3a25
1 parent 2404873
commit 4cf3a25
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/R/tundraContainer-train.R b/R/tundraContainer-train.R
@@ -29,7 +29,7 @@ train <- function(dataframe, train_args = list(), verbose = FALSE, munge = TRUE)
   force(munge)
 
   private$run_hooks("train_pre_munge")
-  if (isTRUE(munge) && length(munge_procedure) > 0) {
+  if (isTRUE(munge) && length(self$.munge_procedure) > 0) {
     dataframe <- munge(dataframe, self$.munge_procedure, verbose)
     attr(dataframe, "mungepieces") <- NULL
   }

diff --git a/tests/testthat/test-train.R b/tests/testthat/test-train.R
@@ -0,0 +1,11 @@
+context("tundraContainer$train")
+
+test_that("it can train a simple example", {
+  container <- tundraContainer$new("foo", function(data) {
+    output$data <- data
+  })
+
+  container$train(iris)
+  expect_identical(container$.output$data, iris)
+})
+