tests/testthat/test-spectral.R

In boupetch/sleepr: Sleep Analysis with R

context("Spectral functions")


test_that("Compute spectral power bands", {
  startTime <- as.POSIXlt(edfReader::readEdfHeader("data/subject1.edf")$startTime, origin = "1970-01-01")
  events <- data.frame(begin = c(startTime,startTime+30,startTime+60),
                  end = c(startTime+30,startTime+60,startTime+90),
                  event = c("N2","N2","REM"))
  events$event <- as.character(events$event)
  write_mdf(edfPath = "data/subject1.edf",
            mdfPath = "data/sample",
            events = events)
  bands_powers <- hypnogram_band_powers(
    record = read_mdf(mdfPath = "data/sample",
                      channels = c("C3-M2"),
                      metadata = FALSE),
    channel = "C3-M2",bands = list(delta = c(0.5,3.5),
                                   theta = c(3.5,8),
                                   alpha = c(8,12),
                                   beta = c(12,30),
                                   gamma1 = c(30,40)))
  expect_equal(nrow(bands_powers), 3)
  expect_equal(length(colnames(bands_powers)), 7)
  
  bands <- list()
  for(i in c(0:39)){
    bands[[paste0(as.character(i),";",as.character(i+1))]] <- c(i,i+1)
  }
  bands_powers <- hypnogram_band_powers(
    record = read_mdf(mdfPath = "data/sample",
                      channels = c("C3-M2"),
                      metadata = FALSE),
    channel = "C3-M2",bands = bands)
  bands_powers <- aggregate_band_powers(bands_powers)
  expect_equal(nrow(bands_powers), 1)
  expect_equal(length(colnames(bands_powers)), 120)
  
  unlink("data/sample",recursive = TRUE)
})


test_that("Compute spectral power bands without REM", {
  startTime <- as.POSIXlt(edfReader::readEdfHeader("data/subject1.edf")$startTime, origin = "1970-01-01")
  events <- data.frame(begin = c(startTime,startTime+30,startTime+60),
                       end = c(startTime+30,startTime+60,startTime+90),
                       event = c("N2","N2","N3"))
  events$event <- as.character(events$event)
  write_mdf(edfPath = "data/subject1.edf",
            mdfPath = "data/sample",
            events = events)
  bands <- list()
  for(i in c(0:39)){
    bands[[paste0(as.character(i),";",as.character(i+1))]] <- c(i,i+1)
  }
  bands_powers <- hypnogram_band_powers(
    record = read_mdf(mdfPath = "data/sample",
                      channels = c("C3-M2"),
                      metadata = FALSE),
    channel = "C3-M2",bands = bands)
  bands_powers <- aggregate_band_powers(bands_powers)
  expect_equal(nrow(bands_powers), 1)
  expect_equal(length(colnames(bands_powers)), 120)
  
  unlink("data/sample",recursive = TRUE)
})

test_that("Compute spectral power bands without NREM", {
  startTime <- as.POSIXlt(edfReader::readEdfHeader("data/subject1.edf")$startTime, origin = "1970-01-01")
  events <- data.frame(begin = c(startTime,startTime+30,startTime+60),
                       end = c(startTime+30,startTime+60,startTime+90),
                       event = c("REM","REM","REM"))
  events$event <- as.character(events$event)
  write_mdf(edfPath = "data/subject1.edf",
            mdfPath = "data/sample",
            events = events)
  bands <- list()
  for(i in c(0:39)){
    bands[[paste0(as.character(i),";",as.character(i+1))]] <- c(i,i+1)
  }
  bands_powers <- hypnogram_band_powers(
    record = read_mdf(mdfPath = "data/sample",
                      channels = c("C3-M2"),
                      metadata = FALSE),
    channel = "C3-M2",bands = bands,resampling = 100)
  bands_powers <- aggregate_band_powers(bands_powers)
  expect_equal(nrow(bands_powers), 1)
  expect_equal(length(colnames(bands_powers)), 40)
  
  unlink("data/sample",recursive = TRUE)
})

test_that("Compute EEMD", {
  startTime <- as.POSIXlt(edfReader::readEdfHeader("data/subject1.edf")$startTime, origin = "1970-01-01")
  events <- data.frame(begin = c(startTime,startTime+30,startTime+60),
                       end = c(startTime+30,startTime+60,startTime+90),
                       event = c("REM","REM","REM"))
  events$event <- as.character(events$event)
  write_mdf(edfPath = "data/subject1.edf",
            mdfPath = "data/sample",
            events = events)
  record <- read_mdf("data/sample",channels = "C3-M2")
  eemd <- hypnogram_eemd(record,"C3-M2",butter = 7)
  expect_equal(length(colnames(eemd)), 33)
  expect_equal(nrow(eemd), 3)
  unlink("data/sample",recursive = TRUE)
})