add testcase to test _human_format function with numbers larger 999T

Signed-off-by: Fabian Degen <[email protected]>

python/interpret-core/tests/visual/test_plot.py

@@ -2,7 +2,7 @@
 # Distributed under the MIT software license
 
 from interpret.visual.plot import plot_line
-
+from interpret.visual.plot import plot_density
 
 def test_plot_line_bounds_smoke():
     data_dict = {
@@ -13,3 +13,18 @@ def test_plot_line_bounds_smoke():
     }
     figure = plot_line(data_dict)
     assert figure.data[0].name == "Lower Bound"
+
+def test_plot_density_large_numbers():
+    """
+    Test that density plots handle large numbers correctly using the new number formatting
+    """
+    data_dict = {
+        "scores": [1.0, 1.0],
+        "names": [9e13, 1e14, 1e15]  # 1e15 value will trigger new formatting
+    }
+
+    figure = plot_density(data_dict)
+
+    # The x-axis tick text should show ranges using our new formatting
+    assert "90T - 100T" in figure.layout.xaxis.ticktext[0]
+    assert "100T - 1.00e+15" in figure.layout.xaxis.ticktext[1]