Experimenting. I expect a linting isue.

Author: jonbrenas

tests/anoph/test_frq_2.py

@@ -0,0 +1,18 @@
+import pytest
+import plotly.graph_objects as go  # type: ignore
+
+
+@pytest.mark.skip
+def test_plot_frequencies_heatmap(api, frq_df):
+    fig = api.plot_frequencies_heatmap(frq_df, show=False, max_len=None)
+    assert isinstance(fig, go.Figure)
+
+    # Test max_len behaviour.
+    with pytest.raises(ValueError):
+        api.plot_frequencies_heatmap(frq_df, show=False, max_len=len(frq_df) - 1)
+
+    # Test index parameter - if None, should use dataframe index.
+    fig = api.plot_frequencies_heatmap(frq_df, show=False, index=None, max_len=None)
+    # Not unique.
+    with pytest.raises(ValueError):
+        api.plot_frequencies_heatmap(frq_df, show=False, index="contig", max_len=None)

tests/anoph/test_snp_frq.py

@@ -12,6 +12,7 @@
 from malariagen_data import ag3 as _ag3
 from malariagen_data.anoph.snp_frq import AnophelesSnpFrequencyAnalysis
 from malariagen_data.util import compare_series_like
+from .test_frq_2 import test_plot_frequencies_heatmap
 
 
 @pytest.fixture
@@ -314,6 +315,8 @@ def test_allele_frequencies_with_str_cohorts(
     # Run the function under test.
     df_snp = api.snp_allele_frequencies(**params)
 
+    test_plot_frequencies_heatmap(api, df_snp)
+
     # Figure out expected cohort labels.
     df_samples = api.sample_metadata(sample_sets=sample_sets)
     if "cohort_" + cohorts in df_samples:
@@ -334,6 +337,8 @@ def test_allele_frequencies_with_str_cohorts(
     # Run the function under test.
     df_aa = api.aa_allele_frequencies(**params)
 
+    test_plot_frequencies_heatmap(api, df_aa)
+
     # Standard checks.
     check_aa_allele_frequencies(
         df=df_aa,