tests/testthat/test-linearAlgebra.R

In ProjectMOSAIC/mosaic: Project MOSAIC Statistics and Mathematics Teaching Utilities

# context('Linear Algebra')


test_that("dot works for vectors", {
  x <- c(1,2,3)
  u <- c(1,1,1)
  expect_equal(ignore_attr = TRUE,  dot(x,u), 6 )
})

test_that("project works for vectors", {
  x <- c(1,2,3)
  u <- c(1,1,1)
  n <- 10
  y <- rnorm(n)
  one <- rep(1,n)
  expect_equal(ignore_attr = TRUE,  project(x, c(1,0,0)), c(1,0,0) ) 
  expect_equal(ignore_attr = TRUE,  project(x, c(0,1,0)), c(0,2,0) )
  expect_equal(ignore_attr = TRUE,  project(x, c(0,0,1)), c(0,0,3) )
  expect_equal(ignore_attr = TRUE,  project(x, u),  c(2,2,2) ) 
  expect_equal(ignore_attr = TRUE,  project(y, one), rep(mean(y), n) )
})

test_that("1 vector works", {
  x <- c(1,2,3)
  u <- c(1,1,1)
  y <- rnorm(10)
  expect_equal(ignore_attr = TRUE,  project(x,1), c(2,2,2) )
  expect_equal(ignore_attr = TRUE,  project(y,1),  rep(mean(y),length(y)) ) 
})

test_that("formula interface to project works in present environment",{
  x <- c(1,2,3)
  b <- c(5,4,3)
  expect_equal( round(project(b~x)[1],4), 1.5714, ignore_attr = TRUE)
  expect_equal( project(b~x+1)[1], 6, ignore_attr = TRUE)
})

test_that("formula interface works with data frame",{
  expect_equal( 
    round(project( wage ~ educ, data=mosaicData::CPS85)[1],3), 
    0.695, ignore_attr = TRUE)
  expect_equal( 
    round(project( wage ~ educ+1, data=mosaicData::CPS85)[1],3), 
    -0.746, ignore_attr = TRUE)
})