scratch/old_code/test--pick.R

In ropenscilabs/taxa: Classes for Storing and Manipulating Taxonomic Data

context("pick")

test_that("pick: ranks", {
  expect_equal(names(ex_hierarchy1$ranklist), c('species', 'genus', 'family'))

  # without piping
  aa1 <- pick(ex_hierarchy1, ranks("family"))
  expect_is(aa1, "Hierarchy")
  expect_equal(names(aa1$ranklist), 'family')
  expect_equal(length(aa1$taxa), 1)

  # with piping
  aa2 <- ex_hierarchy1 %>% pick(ranks("family"))
  expect_is(aa2, "Hierarchy")
  expect_equal(names(aa2$ranklist), 'family')
  expect_equal(length(aa2$taxa), 1)
  expect_equal(aa1, aa2)

  # with piping, many entries
  aa3 <- ex_hierarchy1 %>% pick(ranks("family", "genus"))
  expect_is(aa3, "Hierarchy")
  expect_equal(names(aa3$ranklist), c('family', 'genus'))
  expect_equal(length(aa3$taxa), 2)
})

test_that("pick: names", {
  # without piping
  aa1 <- pick(ex_hierarchy1, nms("Poa"))
  expect_is(aa1, "Hierarchy")
  expect_equal(names(aa1$ranklist), 'genus')
  expect_equal(length(aa1$taxa), 1)

  # with piping
  aa2 <- ex_hierarchy1 %>% pick(nms("Poa"))
  expect_is(aa2, "Hierarchy")
  expect_equal(names(aa2$ranklist), 'genus')
  expect_equal(length(aa2$taxa), 1)
  expect_equal(aa1, aa2)

  # with piping, many entries
  aa3 <- ex_hierarchy1 %>% pick(nms("Poaceae", "Poa"))
  expect_is(aa3, "Hierarchy")
  expect_equal(names(aa3$ranklist), c('family', 'genus'))
  expect_equal(length(aa3$taxa), 2)
})

test_that("pick: ids", {
  # without piping
  aa1 <- pick(ex_hierarchy1, ids(4479))
  expect_is(aa1, "Hierarchy")
  expect_equal(names(aa1$ranklist), 'family')
  expect_equal(length(aa1$taxa), 1)

  # with piping
  aa2 <- ex_hierarchy1 %>% pick(ids(4479))
  expect_is(aa2, "Hierarchy")
  expect_equal(names(aa2$ranklist), 'family')
  expect_equal(length(aa2$taxa), 1)
  expect_equal(aa1, aa2)

  # with piping, many entries
  aa3 <- ex_hierarchy1 %>% pick(ids(4479, 4544))
  expect_is(aa3, "Hierarchy")
  expect_equal(names(aa3$ranklist), c('family', 'genus'))
  expect_equal(length(aa3$taxa), 2)
})


# no variables given to pick - errors
test_that("no variables given to pick - errors", {
  expect_error(pick(ex_hierarchy1), "no acceptable selectors passed in")
})

test_that("pick fails well", {
  expect_error(pick(),
               "argument \".data\" is missing")
  expect_error(pick(5),
               "no 'pick' method for numeric")
  expect_error(pick("adf", 5),
               "no 'pick' method for character")

  expect_error(pick(ex_hierarchy1, ranks()),
               "you must pass in one or more values")
})

test_that("pick: mixed ranks, names and ids", {
  aa1 <- ex_hierarchy1 %>% pick(ranks("family"), ids(4544))

  expect_is(aa1, "Hierarchy")
  expect_equal(names(aa1$ranklist), c('family', 'genus'))
  expect_equal(length(aa1$taxa), 2)
})