gastronomify: Gastronomify a data frame.

library(testthat)
library(plyr)
library(reshape2)
source('../R/gastronomify.r')
# Switch to this: https://github.com/hadley/devtools/wiki/Testing

recipe = c(foo = 2, bar = 1, baz = 1)
fake.recipe.observed = gastronomify(
  x = c(2007, 2007, 2007, 2008, 2008, 2008),
  y = c(8, 9, 6, 2, 12, 5),
  group = factor(c('red', 'blue', 'green', 'red', 'blue',  'green'), levels = c('red', 'blue', 'green')),
  recipe = c(foo = 2, bar = 1, baz = 1)
)
fake.recipe.expected = data.frame(
  row.names = c(2007, 2008),
  foo = c(1.714286, 2.285714),
  bar = c(1.0909091, 0.9090909),
  baz = c(1.6, 0.4)
)
colnames(fake.recipe.expected) <- c('foo (blue)', 'bar (green)',  'baz (red)')

test_that('The result should be structured properly', {
  expect_that(dim(fake.recipe.observed), equals(c(2, 3)))
  expect_that(rownames(fake.recipe.observed), equals(as.character(2007:2008)))
  expect_that(colnames(fake.recipe.observed), equals(c('foo (red)', 'bar (blue)', 'baz (green)')))
})

test_that('The mean recipe should be the base recipe.', {
  expect_that(as.vector(colSums(fake.recipe.observed)), equals(as.vector(recipe * 2)))
})

test_that('This particular recipe should match.', {
  expect_that(rownames(fake.recipe.observed), equals(rownames(fake.recipe.expected)))
# expect_that(round(fake.recipe.observed, -5), equals(round(fake.recipe.expected, -5)))
})

test_that('Fruit salad should work.', {
  recipe =  c(apples = 3, bananas = 1, cherries = 12, grapes = 14,
    kiwis = 2, lemons = 0.5, mangos = 1, nectarines = 2, oranges = 2,
    pineapples = 0.5, raspberries = 8, watermelons = 0.25)
  fruit.salad.observed = gastronomify(
    x = paste('Diet', ChickWeight$Diet),
    y = ChickWeight$weight,
    group = paste(ChickWeight$Time, 'days'),
    recipe = recipe)
  expect_that(as.vector(colMeans(fruit.salad.observed)), equals(as.vector(recipe)))
})

test_that('If there are more items in the recipe than levels, truncate the recipe.', {
  a <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), recipe = c(flour = 2, water = 3))
  b <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), recipe = c(flour = 2, water = 3, oil = 0.5))
  expect_that(a, equals(b))
})

test_that('I should be able to submit a formula rather than separate x and y.', {
  expect_that(1, equals(1), info = 'I should be able to send a formula.')
  expect_that(1, equals(1), info = 'I should be able to send a data parameter with the formula.')
})

test_that('I should be able to specify columns from a data frame rather than separate vectors.', {
  a <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), recipe = c(flour = 2, water = 3))
  b <- gastronomify(x = 'vs', y = 'mpg', group = 'am', data = mtcars, recipe = c(flour = 2, water = 3))
  expect_that(a, equals(b))
})

test_that('The guacamole recipe should be the default', {
# a <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am))
# b <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), guacamole)
# expect_that(a, equals(b))
})


test_that('.inflate should work.', {
  a <- .inflate(iris[-5], 2)
  b <- .inflate(iris[-5], 100)
  print(head(a))
  print(head(b))
  expect_false(all(a == b), info = 'Changing the x should change the result.')
})

test_that('The inflation keyword should work', {
  a <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), inflation = 2)
  b <- gastronomify(x = paste('vs', mtcars$vs), y = mtcars$mpg, group = paste('am', mtcars$am), inflation = 100)
  expect_false(all(a == b), info = 'Changing the inflation parameter should change the result.')
})