Merge pull request #15 from joachim-gassen/master

Attempt at a fix of issue #14
daranzolin · Oct 1, 2019 · 0772271 · 0772271
2 parents 7377b65 + bca43f2
commit 0772271
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/R/viewPipeChain.R b/R/viewPipeChain.R
@@ -49,9 +49,14 @@ processPipeChain <- function(cmd) {
                                           collapse = " ")))
         assign(sprintf("ps%d", i), eval(call))
       }
-      assign(sprintf("obj%d", i), ifelse(is.data.frame(get(sprintf("ps%d", i))),
-                                         as.data.frame(get(sprintf("ps%d", i))),
-                                         get(sprintf("ps%d", i))))
+      assign(
+        sprintf("obj%d", i),
+        if (is.data.frame(get(sprintf("ps%d", i)))) {
+          as.data.frame(get(sprintf("ps%d", i)))
+        } else {
+          get(sprintf("ps%d", i))
+        }
+      )
       if (i == 1 || !identical(get(sprintf("obj%d", i)),
                                get(sprintf("obj%d", i- 1))))
         eval(parse(text = sprintf(cmd, i)))

diff --git a/tools/test_cases.R b/tools/test_cases.R
@@ -50,6 +50,14 @@ iris %>%
             Petal.Length.min = min(Petal.Length),
             Petal.Width.min = min(Petal.Width)) -> min_iris
 
+# Example from issue #14
+
+diamonds %>%
+  select(carat, cut, color, clarity, price) %>%
+  group_by(color) %>%
+  summarise(n = n(), price = mean(price)) %>%
+  arrange(desc(color))
+
 # Assign with pipe (esoteric, works)
 assign("result", mtcars %>% filter(am == 1) %>% select(qsec))