tests/testthat/test-fftw.R

In ravetools: Signal and Image Processing Toolbox for Analyzing Intracranial Electroencephalography Data

require(testthat)
test_that("fftw_r2c", {
  set.seed(1)
  x <- rnorm(1000) + 1
  c <- complex(1000)

  expect_error(ravetools:::fftw_r2c(data = x, HermConj = 1, ret = complex(999)))
  expect_error(ravetools:::fftw_r2c(data = x, HermConj = 0, ret = complex(500)))

  expect_equal(
    ravetools:::fftw_r2c(x, HermConj = 1),
    stats::fft(x)
  )
  expect_equal(
    ravetools:::fftw_r2c(x, HermConj = 0, fftwplanopt = 1),
    stats::fft(x)[1:501]
  )

  # make sure not edit in-place
  set.seed(1)
  xx <- rnorm(1000)
  expect_equal(x, xx + 1)

  c <- complex(1000)
  ravetools:::fftw_r2c(data = x, HermConj = 1, ret = c)
  expect_equal(
    c,
    stats::fft(x)
  )

  expect_equal(
    c[1:501],
    fftwtools::fftw_r2c(x, 0)
  )


  expect_equal(x, xx + 1)


})

test_that("mvfftw_r2c", {
  set.seed(1)
  x <- rnorm(1000)
  dim(x) <- c(100,10)
  c <- apply(x, 2, stats::fft)[1:51,]
  a <- ravetools:::mvfftw_r2c(x, 0)
  cc <- fftwtools::mvfftw_r2c(x, 1)[1:51,]
  expect_equal(a, c)
  expect_equal(a, cc)

  set.seed(1)
  xx <- rnorm(1000)
  dim(xx) <- c(100,10)
  expect_equal(x, xx)

  e <- complex(length(a))
  ravetools:::mvfftw_r2c(x, ret = e)
  expect_equal(e, as.vector(c))

  set.seed(1)
  x <- rnorm(1000)
  dim(x) <- c(100,10)
  b <- ravetools:::mvfftw_r2c(x, 1L)
  # d <- fftwtools::mvfftw_r2c(x, 1, 1)[1:51,]
  expect_equal(b, c)
  # expect_equal(b, d)

  set.seed(1)
  xx <- rnorm(1000)
  dim(xx) <- c(100,10)
  expect_equal(x, xx)
})


test_that("fftw_c2r", {
  set.seed(1)
  x <- rnorm(1000) + 1i * rnorm(1000)
  xx <- x + 1
  c <- double(1000)

  expect_error(ravetools:::fftw_c2r(data = x, HermConj = 1, ret = double(999)))
  expect_error(ravetools:::fftw_c2r(data = x, HermConj = 0, ret = double(1000)))

  expect_equal(x+1, xx)

  expect_equal(
    ravetools:::fftw_c2r(x, HermConj = 1),
    fftwtools::fftw_c2r((xx - 1), HermConj = 1, n = 1000)
  )
  expect_equal(x+1, xx)

  expect_equal(
    ravetools:::fftw_c2r(x, HermConj = 0, fftwplanopt = 1),
    fftwtools::fftw_c2r((xx - 1), HermConj = 0, n = 1998)
  )
  expect_equal(x+1, xx)

  ravetools:::fftw_c2r(data = x, HermConj = 1, ret = c)
  expect_equal(
    c,
    fftwtools::fftw_c2r((xx - 1), HermConj = 1)
  )
  expect_equal(x+1, xx)

  expect_equal(
    c,
    fftwtools::fftw_c2r((xx - 1), HermConj = 1)
  )

  expect_equal(x+1, xx)


})

test_that("fftw_c2c", {
  set.seed(1)
  x <- rnorm(1000) + 1i * rnorm(1000)
  xx <- x + 1
  c <- complex(1000)

  expect_error(ravetools:::fftw_c2c(data = x, ret = double(1000)))
  expect_error(ravetools:::fftw_c2c(data = x, ret = complex(999)))

  expect_equal(x+1, xx)

  expect_equal(
    ravetools:::fftw_c2c(x),
    fftwtools::fftw_c2c((xx - 1))
  )
  expect_equal(x+1, xx)

  expect_equal(
    ravetools:::fftw_c2c(x, fftwplanopt = 1L, ret = c),
    fftwtools::fftw_c2c((xx - 1))
  )
  expect_equal(x+1, xx)

  expect_equal(
    ravetools:::fftw_c2c(x, inverse = TRUE, fftwplanopt = 1L, ret = c),
    fftwtools::fftw_c2c((xx - 1), inverse = TRUE)
  )
  expect_equal(x+1, xx)

  expect_equal(
    c,
    fftwtools::fftw_c2c((xx - 1), inverse = TRUE)
  )

  # inplace with ret == data
  expect_val <- fftwtools::fftw_c2c((xx - 1), inverse = TRUE)
  expect_equal(
    ravetools:::fftw_c2c(x, inverse = TRUE, ret = x),
    expect_val
  )

  expect_equal(x, expect_val)


})