tests/testthat/test-parn88.R

In pmcharrison/incon: Computational Models of Simultaneous Consonance

context("test-encode_pc_set")

test_that("examples", {
  expect_equal(encode_pc_set(c(0, 4, 7)),
               as.integer(c(1, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0)))
  expect_equal(encode_pc_set(c(0)),
               as.integer(c(1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)))
  expect_equal(encode_pc_set(integer()),
               as.integer(c(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)))
})

context("test-pc_weight")

test_that("examples", {
  expect_equal(
    pc_weight(pc = 0,
              pc_set = encode_pc_set(c(0, 4, 7)),
              root_support = root_support_weights$v2),
    10 + 3 + 5
  )
  expect_equal(
    pc_weight(pc = 1,
              pc_set = encode_pc_set(c(0, 4, 7)),
              root_support = root_support_weights$v2),
    0
  )
  expect_equal(
    pc_weight(pc = 2,
              pc_set = encode_pc_set(c(0, 4, 7)),
              root_support = root_support_weights$v2),
    2 + 1
  )
  expect_equal(
    pc_weight(pc = 4,
              pc_set = encode_pc_set(c(0, 4, 7)),
              root_support = root_support_weights$v2),
    10
  )
})

context("test-regression")

library(magrittr)

test_88 <- function(res, ..., digits = 1) {
  expect_equal(
    root_ambiguity(c(...), root_support = "v1") %>% round(digits),
    res
  )
}

test_that("Parncutt (1988): Table 4", {
  # Dyads
  test_88(2.2, 0, 1)
  test_88(2.0, 0, 2)
  test_88(2.1, 0, 3)
  test_88(1.9, 0, 4)
  test_88(1.8, 0, 5)
  test_88(2.2, 0, 6)

  # Triads
  test_88(2.0, 0, 4, 7)
  test_88(2.1, 0, 3, 7)
  test_88(2.3, 0, 4, 8)
  test_88(2.5, 0, 3, 6)

  # Sevenths
  test_88(2.1, 0, 4, 7, 10)
  test_88(2.3, 0, 3, 7, 10)
  test_88(2.3, 0, 4, 7, 11)
  test_88(2.4, 0, 3, 6, 10)
  test_88(2.9, 0, 3, 6, 9)
})

test_that("Sanity checks", {
  expect_equal(root(c(0, 4, 7)), 0)
  expect_equal(root(c(1, 4, 9)), 9)
  expect_gt(root_ambiguity(c(0, 3, 6)),
            root_ambiguity(c(0, 4, 7)))
})

test_that("root_by_pc_chord", {
  chords <- list(
    hrep::pc_chord(c(0, 4, 7)),
    hrep::pc_chord(c(4, 7, 0)),
    hrep::pc_chord(c(7, 2, 5, 11)),
    hrep::pc_chord(c(0, 5, 9))
  )
  chord_ids <- purrr::map_int(chords, hrep::encode_pc_chord)

  root_by_pc_chord[chord_ids] %>%
    expect_equal(c(0, 0, 7, 5))
})