Define all.equal for transformations

CHANGELOG.md

@@ -1,5 +1,6 @@
-### Version 0.1.0.9001
+### Version 0.1.0.9001-2
 
+  * Define `all.equal` for object of S3 class "transformation".
   * Fix a bug in `column_transformation` wherein `NULL` return
     values corrupted the data.frame instead of dropping columns.
 

DESCRIPTION

@@ -6,7 +6,7 @@ Description: A way of thinking about data preparation that
     online prediction so that both can be described by the same codebase.
     With mungebits, you can save time on having to re-implement your
     R code to work in production and instead re-use the same codebase.
-Version: 0.1.0.9001
+Version: 0.1.0.9002
 Author: Robert Krzyzanowski <rob@syberia.io>
 Maintainer: Robert Krzyzanowski <rob@syberia.io>
 Authors@R: c(person("Robert", "Krzyzanowski",

NAMESPACE

@@ -1,5 +1,6 @@
 # Generated by roxygen2 (4.1.1): do not edit by hand
 
+S3method(all,equal.transformation)
 S3method(debug,default)
 S3method(debug,mungebit)
 S3method(debug,mungepiece)

R/column_transformation.R

@@ -309,3 +309,9 @@ print.column_transformation <- function(x, ...) {
   print_transformation(x, ..., byline = "Column transformation")
 }
 
+#' @export
+all.equal.transformation <- function(target, current, ...) {
+  identical(parent.env(environment(target))$transformation,
+            parent.env(environment(current))$transformation)
+}
+

man/mungepiece.Rd

@@ -4,7 +4,7 @@
 \name{mungepiece}
 \alias{mungepiece}
 \title{Mungepiece.}
-\format{\preformatted{Class 'R6ClassGenerator' <environment: 0x7f9a254f7550> 
+\format{\preformatted{Class 'R6ClassGenerator' <environment: 0x7ff39da70188> 
  - attr(*, "name")= chr "mungepiece_generator"
 }}
 \usage{

tests/testthat/test-printing.R

@@ -164,5 +164,16 @@ describe("transformations", {
     ct <- multi_column_transformation(function(x) { 2 * x + 1 }, nonstandard = TRUE)
     expect_output(print(ct), "non-standard", fixed = TRUE)
   })
+
+  test_that("it can tell when the train and predict of a column transformation differ", {
+    bit <- mungebit$new(column_transformation(function(x) x), column_transformation(function(x) x + 1))
+    expect_output(print(bit), "train function")
+    expect_output(print(bit), "predict function")
+  })
+
+  test_that("it can tell when the train and predict of a column transformation are equivalent", {
+    bit <- mungebit$new(column_transformation(function(x) x))
+    expect_output(print(bit), "train and predict function")
+  })
 })