test_that("print.satab prints to console", {
	obj <- satab(outdesign_crd_satab)
	expect_s3_class(obj, "satab")

	expect_invisible(print(obj))
	expect_output(print(obj), "Source of Variation")
	expect_output(print(obj), "Residual")
})

test_that("satab examples print expected sections", {
	expect_output(print(satab(outdesign_crd_satab)), "Source of Variation")
	expect_output(print(satab(outdesign_rcbd_satab)), "Block stratum")
	expect_output(print(satab(outdesign_lsd_satab)), "Row")

	expect_output(print(satab(outdesign_crossed_satab)), "Residual")
	expect_output(print(satab(outdesign_crossed_rcbd_satab)), "Residual")
	expect_output(print(satab(outdesign_crossed_lsd_satab)), "Residual")

	expect_output(print(satab(outdesign_nested_satab)), "Column")
	expect_output(print(satab(outdesign_split_satab)), "Whole plot Residual")
})

test_that("get_anova_structure throws error for unknown design type", {
	# Create a minimal design book
	design_book <- data.frame(
		plot = 1:4,
		treatment = c("A", "B", "A", "B")
	)

	# Test that unknown design type triggers error
	expect_error(
		biometryassist:::get_anova_structure("unknown_design", design_book),
		"Unknown design type: unknown_design"
	)

	expect_error(
		biometryassist:::get_anova_structure("invalid", design_book),
		"Unknown design type: invalid"
	)
})

test_that("anova_split handles different subplot column names", {
	# Test with 'subplots' column name
	design_book_subplots <- data.frame(
		plots = 1:8,
		block = rep(1:2, each = 4),
		wholeplots = rep(1:4, each = 2),
		subplots = rep(1:2, 4),
		TreatmentA = rep(c("A1", "A2"), each = 4),
		TreatmentB = rep(c("B1", "B2"), 4)
	)

	result1 <- biometryassist:::anova_split(design_book_subplots)
	expect_type(result1, "list")
	expect_true("sources" %in% names(result1))
	expect_true("df" %in% names(result1))

	# Test with 'splots' column name
	design_book_splots <- data.frame(
		plots = 1:8,
		block = rep(1:2, each = 4),
		wholeplots = rep(1:4, each = 2),
		splots = rep(1:2, 4),
		TreatmentA = rep(c("A1", "A2"), each = 4),
		TreatmentB = rep(c("B1", "B2"), 4)
	)

	result2 <- biometryassist:::anova_split(design_book_splots)
	expect_type(result2, "list")
	expect_true("sources" %in% names(result2))
	expect_true("df" %in% names(result2))

	# Results should be identical (same df structure)
	expect_equal(result1$df, result2$df)
})

test_that("anova_split throws error when subplot column is missing", {
	# Design book without subplots or splots column
	design_book_invalid <- data.frame(
		plots = 1:8,
		block = rep(1:2, each = 4),
		wholeplots = rep(1:4, each = 2),
		TreatmentA = rep(c("A1", "A2"), each = 4),
		TreatmentB = rep(c("B1", "B2"), 4)
	)

	expect_error(
		biometryassist:::anova_split(design_book_invalid),
		"Cannot find subplot column in design book"
	)
})

test_that("anova_split throws error when wrong number of treatment columns", {
	# Design book with only 1 treatment column
	design_book_one_trt <- data.frame(
		plots = 1:8,
		block = rep(1:2, each = 4),
		wholeplots = rep(1:4, each = 2),
		subplots = rep(1:2, 4),
		TreatmentA = rep(c("A1", "A2"), each = 4)
	)

	expect_error(
		biometryassist:::anova_split(design_book_one_trt),
		"Expected 2 treatment columns in split plot design, found 1"
	)

	# Design book with 3 treatment columns
	design_book_three_trt <- data.frame(
		plots = 1:8,
		block = rep(1:2, each = 4),
		wholeplots = rep(1:4, each = 2),
		subplots = rep(1:2, 4),
		TreatmentA = rep(c("A1", "A2"), each = 4),
		TreatmentB = rep(c("B1", "B2"), 4),
		TreatmentC = rep(c("C1", "C2"), 4)
	)

	expect_error(
		biometryassist:::anova_split(design_book_three_trt),
		"Expected 2 treatment columns in split plot design, found 3"
	)
})

test_that("anova_strip throws error when block column is missing", {
	design_book_missing_block <- data.frame(
		plots = 1:4,
		wholeplots = 1:4,
		subplots = rep(1:2, 2),
		TreatmentA = rep(c("A1", "A2"), each = 2),
		TreatmentB = rep(c("B1", "B2"), 2)
	)

	expect_error(
		biometryassist:::anova_strip(design_book_missing_block),
		"Expected a 'block' column in strip plot design"
	)
})

test_that("anova_strip throws error when wrong number of treatment columns", {
	# Only 1 treatment column
	design_book_one_trt <- data.frame(
		plots = 1:4,
		block = rep(1:2, each = 2),
		wholeplots = 1:4,
		subplots = rep(1:2, 2),
		TreatmentA = rep(c("A1", "A2"), each = 2)
	)

	expect_error(
		biometryassist:::anova_strip(design_book_one_trt),
		"Expected 2 treatment columns in strip plot design, found 1"
	)

	# 3 treatment columns
	design_book_three_trt <- data.frame(
		plots = 1:4,
		block = rep(1:2, each = 2),
		wholeplots = 1:4,
		subplots = rep(1:2, 2),
		TreatmentA = rep(c("A1", "A2"), each = 2),
		TreatmentB = rep(c("B1", "B2"), 2),
		TreatmentC = rep(c("C1", "C2"), 2)
	)

	expect_error(
		biometryassist:::anova_strip(design_book_three_trt),
		"Expected 2 treatment columns in strip plot design, found 3"
	)
})