tests/testthat/test-read_srt.R

In benjcunningham/subtitler: Read Subtitle Files

context("Work with SubRip Files")

suppressMessages({

  library(readr)

  srt <- read_srt("Always_Sunny_S10E04.srt")
  csv <- read_csv("Always_Sunny_S10E04.csv")

  cols <- c("index", "start", "end", "text")

})


test_that("Read in data frame looks as expected", {

  expect_equal(srt[cols], csv[cols])

})

test_that("Timestamps convert to milliseconds", {

  expect_equal(as_milliseconds(srt$start), csv$start_ms)
  expect_equal(as_milliseconds(srt$end), csv$end_ms)

})

test_that("Milliseconds convert to timestamps", {

  expect_equal(as_timestamp(csv$start_ms), srt$start)
  expect_equal(as_timestamp(csv$end_ms), srt$end)

})

test_that("Timestamps can be conveniently synchronized", {

  expect_equal(
    as_milliseconds(add_milliseconds(srt$start, 100)),
    as_milliseconds(srt$start) + 100
  )
  expect_equal(
    as_milliseconds(add_milliseconds(srt$start, -10)),
    as_milliseconds(srt$start) - 10
  )
  expect_equal(
    as_milliseconds(add_milliseconds(srt$start, 1:nrow(srt))),
    as_milliseconds(srt$start) + 1:nrow(srt)
  )

})

test_that("Written out SRT looks exactly like the original", {

  tmp <- tempfile()
  on.exit(unlink(tmp))

  write_srt(srt, tmp)

  infile <- read_file("Always_Sunny_S10E04.srt")
  outfile <- read_file(tmp)

  expect_equal(infile, outfile)

})