From 56d83d9292244f2d3891b62cd66912513028a04a Mon Sep 17 00:00:00 2001
From: dtm2451 <daniel.bunis@ucsf.edu>
Date: Tue, 2 Jan 2024 17:34:57 -0500
Subject: [PATCH] test expanded plotScoreHeatmap color control

---
 tests/testthat/test-heatmap.R | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/testthat/test-heatmap.R b/tests/testthat/test-heatmap.R
index 732c166..4a36817 100644
--- a/tests/testthat/test-heatmap.R
+++ b/tests/testthat/test-heatmap.R
@@ -63,12 +63,17 @@ test_that("heatmap - can pass excess pheatmap::pheatmap parameters through plotS
         5)
 })
 
-test_that("heatmap scores color can be adjusted when 'normalize = FALSE'", {
+test_that("heatmap scores color can be adjusted, regardless of 'normalize' value", {
     expect_equal(
         plotScoreHeatmap(results = pred, silent = TRUE, return.data = TRUE,
             normalize = FALSE,
             color = colorRampPalette(c("red", "blue"))(33))$color,
         colorRampPalette(c("red", "blue"))(33))
+    expect_equal(
+        plotScoreHeatmap(results = pred, silent = TRUE, return.data = TRUE,
+            normalize = TRUE,
+            color = colorRampPalette(c("red", "blue"))(33))$color,
+        colorRampPalette(c("red", "blue"))(33))
 })
 
 test_that("heatmap is adjusted properly when 'labels.use' yields 1 or 0 labels", {