Merge pull request #41 from jennybc/master

add test that [ does not change class
tidyverse · Mar 17, 2016 · 77f2d47 · 77f2d47
2 parents 88f86cd + b15609a
commit 77f2d47
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/testthat/test-tbl-df.r b/tests/testthat/test-tbl-df.r
@@ -7,6 +7,13 @@ test_that("[ never drops", {
   expect_equal(mtcars2[, 1], mtcars2[1])
 })
 
+test_that("[ retains class", {
+  mtcars2 <- tbl_df(mtcars)
+  expect_identical(class(mtcars2), class(mtcars2[1:5, ]))
+  expect_identical(class(mtcars2), class(mtcars2[, 1:5]))
+  expect_identical(class(mtcars2), class(mtcars2[1:5, 1:5]))
+})
+
 test_that("[ with 0 cols creates correct row names (#656)", {
   zero_row <- tbl_df(iris)[, 0]
   expect_is(zero_row, "tbl_df")