test_that("add_radii_of_gyration() works", {
  in_df <- mp_table[which(!is.na(mp_table$mass))[1:20], ]
  out_df <- add_radii_of_gyration(in_df)
  in_df$kx <- sqrt(in_df$Ixx / in_df$mass)
  in_df$ky <- sqrt(in_df$Iyy / in_df$mass)
  in_df$kz <- sqrt(in_df$Izz / in_df$mass)

  expect_true(all.equal(out_df, in_df))
})

test_that("rollup_radii_of_gyration_unc() works", {
  in_df <- add_radii_of_gyration(rollup_mass_props_and_unc(sawe_tree, sawe_table))
  out_df <- rollup_radii_of_gyration_unc(sawe_tree, in_df)
  part <- which(in_df$id != "Combined")
  in_part_df <- in_df[part, ]
  comb <- which(in_df$id == "Combined")
  in_comb_df <- in_df[comb, ]
  corr <- list(
    x = sum(in_part_df$sigma_mass^2 * ((in_part_df$Cy - in_comb_df$Cy)^2 + (in_part_df$Cz - in_comb_df$Cz)^2)),
    y = sum(in_part_df$sigma_mass^2 * ((in_part_df$Cx - in_comb_df$Cx)^2 + (in_part_df$Cz - in_comb_df$Cz)^2)),
    z = sum(in_part_df$sigma_mass^2 * ((in_part_df$Cx - in_comb_df$Cx)^2 + (in_part_df$Cy - in_comb_df$Cy)^2))
  )
  in_df$sigma_kx <- sqrt(in_df$sigma_Ixx^2 / (4 * in_df$mass * in_df$Ixx) + (in_df$Ixx * in_df$sigma_mass^2) / (4 * in_df$mass^3))
  in_df$sigma_ky <- sqrt(in_df$sigma_Iyy^2 / (4 * in_df$mass * in_df$Iyy) + (in_df$Iyy * in_df$sigma_mass^2) / (4 * in_df$mass^3))
  in_df$sigma_kz <- sqrt(in_df$sigma_Izz^2 / (4 * in_df$mass * in_df$Izz) + (in_df$Izz * in_df$sigma_mass^2) / (4 * in_df$mass^3))

  expect_true(all.equal(out_df[part], in_df[part]))
  expect_equal(out_df[comb, "sigma_kx"], sqrt(in_df[comb, "sigma_kx"]^2 - corr$x / (2 * in_df[comb, "mass"]^2)))
  expect_equal(out_df[comb, "sigma_ky"], sqrt(in_df[comb, "sigma_ky"]^2 - corr$y / (2 * in_df[comb, "mass"]^2)))
  expect_equal(out_df[comb, "sigma_kz"], sqrt(in_df[comb, "sigma_kz"]^2 - corr$z / (2 * in_df[comb, "mass"]^2)))
})

test_that("get_mass_props_and_unc_and_radii() works", {
  in_df <- mp_table[which(!is.na(mp_table$mass))[1:20], ]
  out_df <- add_radii_of_gyration(in_df)
  expected <- out_df[which(out_df$id == "C.1.1.1.1.1.1.1"), ]
  actual <- get_mass_props_and_unc_and_radii(out_df, "C.1.1.1.1.1.1.1")

  expect_equal(unname(actual$mass), unname(expected$mass))
  expect_equal(unname(actual$radii_gyration["x"]), unname(expected$kx))
  expect_equal(unname(actual$radii_gyration["y"]), unname(expected$ky))
  expect_equal(unname(actual$radii_gyration["z"]), unname(expected$kz))
})

test_that("get_mass_props_and_unc_and_radii_and_unc() works", {
  in_df <- add_radii_of_gyration(sawe_table)
  out_df <- rollup_radii_of_gyration_unc(sawe_tree, in_df)
  expected <- out_df[which(out_df$id == "Combined"), ]
  actual <- get_mass_props_and_unc_and_radii_and_unc(out_df, "Combined")

  expect_equal(unname(actual$mass), unname(expected$mass))
  expect_equal(unname(actual$radii_gyration["x"]), unname(expected$kx))
  expect_equal(unname(actual$radii_gyration["y"]), unname(expected$ky))
  expect_equal(unname(actual$radii_gyration["z"]), unname(expected$kz))
  expect_equal(unname(actual$sigma_radii_gyration["x"]), unname(expected$sigma_kx))
  expect_equal(unname(actual$sigma_radii_gyration["y"]), unname(expected$sigma_ky))
  expect_equal(unname(actual$sigma_radii_gyration["z"]), unname(expected$sigma_kz))
})

test_that("set_radii_of_gyration_unc() works", {
  mp <- list(sigma_radii_gyration = c(x = 31, y = 59, z = 83))
  actual <- set_radii_of_gyration_unc(mp_table, "C.1", mp)
  row <- which(actual$id == "C.1")

  expect_equal(unname(actual[row, "sigma_kx"]), unname(mp$sigma_radii_gyration["x"]))
  expect_equal(unname(actual[row, "sigma_ky"]), unname(mp$sigma_radii_gyration["y"]))
  expect_equal(unname(actual[row, "sigma_kz"]), unname(mp$sigma_radii_gyration["z"]))
})