From 7b69da9074aa6eec0110a658feda611accdcd6da Mon Sep 17 00:00:00 2001
From: jonbrenas <51911846+jonbrenas@users.noreply.github.com>
Date: Thu, 12 Dec 2024 10:28:18 +0000
Subject: [PATCH] Test

---
 tests/anoph/test_hap_frq.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/anoph/test_hap_frq.py b/tests/anoph/test_hap_frq.py
index b8278aa9..165eb958 100644
--- a/tests/anoph/test_hap_frq.py
+++ b/tests/anoph/test_hap_frq.py
@@ -9,10 +9,10 @@
 from malariagen_data import ag3 as _ag3
 from malariagen_data.anoph.hap_frq import AnophelesHapFrequencyAnalysis
 from .test_frq import (
+    test_plot_frequencies_heatmap,
     test_plot_frequencies_time_series,
     test_plot_frequencies_time_series_with_taxa,
     test_plot_frequencies_time_series_with_areas,
-    test_plot_frequencies_interactive_map,
 )
 
 
@@ -91,7 +91,7 @@ def check_hap_frequencies_advanced(
     test_plot_frequencies_time_series(api, ds)
     test_plot_frequencies_time_series_with_taxa(api, ds)
     test_plot_frequencies_time_series_with_areas(api, ds)
-    test_plot_frequencies_interactive_map(api, ds)
+    # test_plot_frequencies_interactive_map(api, ds)
     assert set(ds.dims) == {"cohorts", "variants"}
 
     expected_cohort_vars = [
@@ -164,6 +164,8 @@ def test_hap_frequencies_with_str_cohorts(
     # Run the function under test.
     df_hap = api.haplotypes_frequencies(**params)
 
+    test_plot_frequencies_heatmap(api, df_hap)
+
     # Standard checks.
     check_hap_frequencies(
         api=api,