context("Basic Matlab functions")

test_that("rand works properly", {
    expect_equal(dim(rand()), c(1, 1))
    expect_equal(dim(rand(1, 2)), c(1, 2))
    expect_equal(dim(rand(3, 2)), c(3, 2))
})

test_that("repmat works properly", {
    mx0 <- c(1:4) # when converted to matrix, results in a column vector
    mx1 <- matrix(5:8)
    mx2 <- matrix(0:-3, 2)
    expect_error(repmat(mx0))
    expect_equal(repmat(mx0, 1), as.matrix(mx0))
    expect_equal(
        object = repmat(mx0, 2),
        expected = unname(t(cbind(rbind(mx0, mx0), rbind(mx0, mx0))))
    )
    expect_equal(
        object = repmat(mx1, 2),
        expected = unname(cbind(rbind(mx1, mx1), rbind(mx1, mx1)))
    )
    expect_equal(
        object = repmat(mx2, c(2, 3)),
        expected = cbind(rbind(mx2, mx2), rbind(mx2, mx2), rbind(mx2, mx2))
    )
    expect_equal(
        object = repmat(mx2, c(4, 1)),
        expected = rbind(mx2, mx2, mx2, mx2)
    )
})

test_that("zeros and ones work as expected", {
    expect_equal(zeros(1), matrix(0, 1))
    expect_equal(zeros(2), matrix(0, 2, 2))
    expect_equal(zeros(2, 1), matrix(0, 2, 1))
    expect_equal(zeros(1, 10), matrix(0, 1, 10))
    expect_equal(ones(8), matrix(1, 8, 8))
    expect_equal(ones(5, 2), matrix(1, 5, 2))
    expect_equal(ones(2, 100), matrix(1, 2, 100))
})

test_that("times works as expected", {
    expect_equal(times(9, 6), as.matrix(54))
    expect_equal(times(9, c(.8, 9)), as.matrix(c(7.2, 81)))
    expect_equal(times(c(.8, 9), 5), as.matrix(c(4, 45)))
    expect_equal(times(matrix(1:4, 2), 5), matrix(c(5, 10, 15, 20), 2))
    expect_equal(times(5, matrix(1:4, 2)), matrix(c(5, 10, 15, 20), 2))
    expect_equal(times(matrix(1:4, 2), c(10, 3)), matrix(c(10, 6, 30, 12), 2))
    expect_equal(
        object = times(matrix(1:4, 2), matrix(c(10, 3), 1)),
        expected = matrix(c(10, 20, 9, 12), 2)
    )
    expect_equal(times(c(10, 3), matrix(1:4, 2)), matrix(c(10, 6, 30, 12), 2))
    expect_equal(
        object = times(matrix(c(10, -5, 3, 9), 2), matrix(1:4, 2)),
        expected = matrix(c(10, -10, 9, 36), 2)
    )
    expect_equal(
        object = times(matrix(c(-1.6, 5), 1), c(8, 1)),
        expected = matrix(c(-12.8, -1.6, 40, 5), 2)
    )
})

test_that("colon works as expected (hee hee)", {
    expect_equal(colon(1, 4), 1:4)
    expect_length(colon(4, 1), 0)
})

test_that("size works as on MATLAB", {
    sk <- 10
    vk <- 1:4
    mx <- matrix(1:6, 2)
    ra <- array(1:24, c(2, 3, 4))
    expect_equal(size(sk), 1)
    expect_equal(size(vk), c(1, 4))
    expect_equal(size(mx), c(2, 3))
    expect_equal(size(ra), c(2, 3, 4))
    expect_equal(size(sk, 199), 1)
    expect_equal(size(vk, 199), 1)
    expect_equal(size(mx, 199), 1)
    expect_equal(size(ra, 199), 1)
    expect_equal(size(vk, 2), 4)
    expect_equal(size(mx, 2), 3)
    expect_equal(size(ra, 2), 3)
    expect_equal(size(ra, 3), 4)
})