Add unit test to make sure austen_books() has right factor levels

juliasilge · Oct 23, 2016 · 3f12439 · 3f12439
1 parent 6bd90d0
commit 3f12439
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tests/testthat.R b/tests/testthat.R
@@ -0,0 +1,4 @@
+library(testthat)
+library(janeaustenr)
+
+test_check("janeaustenr")
diff --git a/tests/testthat/test_austen_books.R b/tests/testthat/test_austen_books.R
@@ -0,0 +1,15 @@
+# tests for austen_books function
+
+context("Tidy dataframe for books")
+
+suppressPackageStartupMessages(library(dplyr))
+
+test_that("tidy frame for Austen books is right", {
+        d <- austen_books() %>% 
+                group_by(book) %>%
+                summarise(total_lines = n())
+        expect_equal(nrow(d), 6)
+        expect_equal(ncol(d), 2)
+        expect_equal(as.character(d$book[1]), "Sense & Sensibility")
+        expect_equal(as.character(d$book[6]), "Persuasion")
+})