#' Copyright(c) 2017-2023 R. Mark Sharp
#' This file is part of nprcgenekeepr
context("makeSimPed")
library(testthat)

ped <- nprcgenekeepr::lacy1989Ped
# nolint start: object_name_linter.
simParent_1 <- list(id = "A",
                    sires = c("s1_1", "s1_2", "s1_3"),
                    dams = c("d1_1", "d1_2", "d1_3", "d1_4"))
simParent_2 <- list(id = "B",
                    sires = c("s2_1", "s2_2", "s2_3"),
                    dams = c("d2_1", "d2_2", "d2_3", "d2_4"))
simParent_3 <- list(id = "E",
                    sires = c("s3_1", "s3_2", "s3_3"),
                    dams = c("d3_1", "d3_2", "d3_3", "d3_4"))
allSimParents <- list(simParent_1, simParent_2, simParent_3)

set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents)

test_that("makeSimPed creates a correct pedigree structure", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_equal(simPed$sire[simPed$id == "B"], "s2_2")
  expect_equal(simPed$sire[simPed$id == "E"], "s3_1")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_equal(simPed$dam[simPed$id == "B"], "d2_4")
  expect_equal(simPed$dam[simPed$id == "E"], "d3_4")
})

set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents, verbose = TRUE)

test_that("makeSimPed creates a correct pedigree structure", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_equal(simPed$sire[simPed$id == "B"], "s2_2")
  expect_equal(simPed$sire[simPed$id == "E"], "s3_1")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_equal(simPed$dam[simPed$id == "B"], "d2_4")
  expect_equal(simPed$dam[simPed$id == "E"], "d3_4")
})

simParent_4 <- list(id = "B",
                    sires = NULL,
                    dams = c("d2_1", "d2_2", "d2_3", "d2_4"))
allSimParents <- list(simParent_1, simParent_4, simParent_3)
set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents)

test_that("makeSimPed creates a correct pedigree structure with no sires", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_true(is.na(simPed$sire[simPed$id == "B"]))
  expect_equal(simPed$sire[simPed$id == "E"], "s3_3")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_equal(simPed$dam[simPed$id == "B"], "d2_3")
  expect_equal(simPed$dam[simPed$id == "E"], "d3_1")
})

set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents, verbose = TRUE)

test_that("makeSimPed creates a correct pedigree structure with no sires", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_true(is.na(simPed$sire[simPed$id == "B"]))
  expect_equal(simPed$sire[simPed$id == "E"], "s3_3")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_equal(simPed$dam[simPed$id == "B"], "d2_3")
  expect_equal(simPed$dam[simPed$id == "E"], "d3_1")
})

simParent_5 <- list(id = "B",
                    sires = c("s2_1", "s2_2", "s2_3"),
                    dams = NULL)

allSimParents <- list(simParent_1, simParent_5, simParent_3)
set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents)

test_that("makeSimPed creates a correct pedigree structure with no dams", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_equal(simPed$sire[simPed$id == "B"], "s2_2")
  expect_equal(simPed$sire[simPed$id == "E"], "s3_3")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_true(is.na(simPed$dam[simPed$id == "B"]))
  expect_equal(simPed$dam[simPed$id == "E"], "d3_1")
})

set_seed(seed = 1)

simPed <- makeSimPed(ped, allSimParents, verbose = TRUE)

test_that("makeSimPed creates a correct pedigree structure with no dams", {
  expect_equal(simPed$sire[simPed$id == "A"], "s1_1")
  expect_equal(simPed$sire[simPed$id == "B"], "s2_2")
  expect_equal(simPed$sire[simPed$id == "E"], "s3_3")

  expect_equal(simPed$dam[simPed$id == "A"], "d1_2")
  expect_true(is.na(simPed$dam[simPed$id == "B"]))
  expect_equal(simPed$dam[simPed$id == "E"], "d3_1")
  # nolint end: object_name_linter.
})