tests/testthat/test_weighted_line.R

In martinSter/robsurvey: Robust Survey Statistics Estimation

context("weighted_line")
library(robsurvey)

test_that("Function outputs correct result for example", {
  data(cars)
  out1 <- weighted_line(cars$speed, cars$dist, w=rep(1, length(cars$speed)))
  out2 <- weighted_line(cars$speed, cars$dist, w=rep(1:10, each=5))
  expect_equal(coef(out1), c(-29.333333, 4.666667), tolerance=1e-05)
  expect_equal(coef(out2), c(-17.111111, 3.777778), tolerance=1e-05)
})

test_that("weighted_line() results are identical to line() results", {
  data(cars)
  out1 <- weighted_line(cars$speed, cars$dist, w=rep(1, length(cars$speed)))
  out2 <- line(cars$speed, cars$dist)
  expect_equal(coef(out1), coef(out2), tolerance=1e-05)
})