tests/testthat/test_pelleted.R

In wfondrie/evrspin: Calculations for Extracellular Vesicle Sedimentation by Centrifugation

library(evrspin)
context("Fraction pelleted calculations")

sw <- rotor("sw40")
fa <- rotor("type60")

test_that("Errors are working", {
  # t
  expect_error(pelleted(t = "a", 1, 1, sw), "t must be numeric")
  expect_error(pelleted(t = -1, 1, 1, sw), "t must be numeric")
  expect_error(pelleted(t = -1:10, 1, 1, sw), "t must be numeric")

  # d
  expect_error(pelleted(1, d = "a", 1, sw), "d must be numeric")
  expect_error(pelleted(1, d = -1, 1, sw), "d must be numeric")
  expect_error(pelleted(1, d = -1:10, 1, sw), "d must be numeric")

  # rcf
  expect_error(pelleted(1, 1, rcf = "a", sw), "rcf must be numeric")
  expect_error(pelleted(1, 1, rcf = -1, sw), "rcf must be numeric")
  expect_error(pelleted(1, 1, rcf = -1:10, sw), "rcf must be numeric")

  # rotor
  expect_error(pelleted(1, 1, 1, rotor = "a"), "rotor is not a ")

  # vesicle and solvent densities
  expect_error(pelleted(1, 1, 1, sw, vesicle.density = "a"), "vesicle.density must be numeric.")
  expect_error(pelleted(1, 1, 1, sw, solvent.density = "a"), "solvent.density must be numeric.")
  expect_error(pelleted(1, 1, 1, sw, vesicle.density = 1, solvent.density = 2), "vesicle.density must be greater")

  # viscosity
  expect_error(pelleted(1, 1, 1, sw, viscosity = "a"), "viscosity must be numeric")
  expect_error(pelleted(1, 1, 1, sw, viscosity = -1), "viscosity must be numeric")
  expect_error(pelleted(1, 1, 1, sw, viscosity = -1:10), "viscosity must be numeric")
})

test_that("Output is a numeric vector of correct length", {
  expect_match(class(pelleted(1, 1, 1, sw)), "numeric")
  expect_equal(length(pelleted(1, 1, 1, sw)), 1)

  expect_match(class(pelleted(1:10, 1, 1, sw)), "numeric")
  expect_equal(length(pelleted(1:10, 1, 1, sw)), 10)

  expect_match(class(pelleted(1:10, 1:10, 1, sw)), "numeric")
  expect_equal(length(pelleted(1:10, 1:10, 1, sw)), 10)

  expect_match(class(pelleted(1, 1, 1, fa)), "numeric")
  expect_equal(length(pelleted(1, 1, 1, fa)), 1)

  expect_match(class(pelleted(1:10, 1, 1, fa)), "numeric")
  expect_equal(length(pelleted(1:10, 1, 1, fa)), 10)

  expect_match(class(pelleted(1:10, 1:10, 1, fa)), "numeric")
  expect_equal(length(pelleted(1:10, 1:10, 1, fa)), 10)
})

test_that("Calculations are correct", {
  expect_equal(round(pelleted(30, 150, 10000, sw), 2), 0.30)
  expect_equal(round(pelleted(30, 150, 10000, fa), 2), 0.88)
})