tests/testthat/test-predict.R

In andreabecsek/NaiveBayes: Gaussian Naive Bayes model

context('predictions')
library(MASS)

test_that("predictions are calculated correctly",{
  # some random data
  # observations in class 1
  mu_1 <- c(40,-80)
  sd_1 <- c(1,1)
  C1 <- mvrnorm(60,mu_1, diag(sd_1^2))
  # observations in class 2
  mu_2 <- c(-50,60)
  sd_2 <- c(1,1)
  C2 <- mvrnorm(40,mu_2, diag(sd_2^2))
  # full data matrix
  X <- rbind(C1,C2)
  y <- matrix(c(rep(0,60),rep(1,40)),ncol=1)
  # fit model
  model <- naive_bayes(X,y)
  # generate test data
  C1_test <- mvrnorm(5,mu_1, diag(sd_1^2))
  C2_test <- mvrnorm(5,mu_2, diag(sd_2^2))
  X_test <- rbind(C1_test,C2_test)
  y_test <- matrix(c(rep(0,5),rep(1,5)),ncol=1)
  # make predictions
  pred <- predict(model,X_test)
  expect_equal(pred,y_test)
})