DS-3563: Protect Sum against NULL in print method (#6)

* DS-3563 Simplify and protect against NULL [revdep skip]

Ensure  custom.color set as NULL is handled

Co-authored-by: jrwishart <justin.wishart@displayr.com>

DESCRIPTION

@@ -27,5 +27,5 @@ Remotes: Displayr/flipData,
  Displayr/flipU,
  Displayr/rhtmlSankeyTree,
  Displayr/verbs
-RoxygenNote: 7.1.1
+RoxygenNote: 7.1.2
 Encoding: UTF-8

R/cart.R

@@ -396,7 +396,7 @@ print.CART <- function(x, ...)
     if (x$output == "Sankey")
     {
         frame <- rPartToTreeFrame(x)
-        tree.list <- treeFrameToList(frame, attr(x, "xlevels"), x$model, x$where, x$labels)
+        tree.list <- treeFrameToList(frame, attr(x, "xlevels"), x$model, x$where, x$labels, ...)
         plt <- SankeyTree(tree.list, value = "n", nodeHeight = 100, numeric.distribution = TRUE,
                           tooltip = "tooltip", treeColors = TRUE, terminalDescription = TRUE)
         print(plt)

R/sankey.R

@@ -112,6 +112,9 @@ treeFrameToList <- function(frame, xlevels, model, assigned, labels, max.tooltip
         })
     }
 
+    if (length(custom.color) == 0L)
+        custom.color <- formals()[["custom.color"]]
+
     if (outcome.is.factor)
     { # Classification tree.
         tree.type = "Classification"

tests/testthat/test-CART.R

@@ -19,6 +19,8 @@ test_that("saving variables",
                                    data = bank, subset = bank$ID > 100))
         expect_error(predict(z), NA)
         expect_error(flipData::Probabilities(z), NA)
+        expect_error(print(z), NA)
+        expect_error(print(z, custom.color = NULL), NA)
     })