WIP: plot testing for all float types

observingClouds · Dec 15, 2023 · 33bc0c6 · 33bc0c6
1 parent 1470a76
commit 33bc0c6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/test_visualisation.py b/tests/test_visualisation.py
@@ -3,7 +3,7 @@
 import xarray as xr
 
 import xbitinfo as xb
-from xbitinfo.graphics import add_bitinfo_labels
+from xbitinfo.graphics import add_bitinfo_labels, plot_bitinformation
 
 
 def test_add_bitinfo_labels():
@@ -50,3 +50,9 @@ def test_add_bitinfo_labels():
             assert ax.texts[i + 5].get_text() == keepbits_text
     # Cleanup the plot
     plt.close()
+
+@pytest.mark.parametrize("dtype", ["float64", "float32", "float16"])
+def test_plot_bitinformation(rasm, dtype)
+    ds = rasm.astype(dtype)
+    info_per_bit = xb.get_bitinformation(ds, dim="lon")
+    plot_bitinformation(info_per_bit)