test that table uses repr and doesn't try to convert unitized data

story645 · rcomer · story645 · commit b40a67c34bcd · 2024-01-24T14:01:15.000-05:00
Co-authored-by: Elliott Sales de Andrade &lt;quantum.analyst@gmail.com&gt;

Co-authored-by: Ruth Comer &lt;10599679+rcomer@users.noreply.github.com&gt;
diff --git a/lib/matplotlib/tests/test_table.py b/lib/matplotlib/tests/test_table.py
@@ -1,3 +1,6 @@
+import datetime
+from unittest.mock import Mock
+
 import numpy as np
 import pytest
 
@@ -7,6 +10,7 @@
 from matplotlib.table import CustomCell, Table
 from matplotlib.testing.decorators import image_comparison, check_figures_equal
 from matplotlib.transforms import Bbox
+import matplotlib.units as munits
 
 
 def test_non_square():
@@ -229,3 +233,34 @@ def test_table_bbox(fig_test, fig_ref):
                   loc='center',
                   bbox=Bbox.from_extents(0.1, 0.2, 0.9, 0.8)
                   )
+
+
+@check_figures_equal(extensions=['png'])
+def test_table_unit(fig_test, fig_ref):
+    # test that table doesn't participate in unit machinery, instead uses repr/str
+
+    class FakeUnit:
+        def __init__(self, thing):
+            pass
+        def __repr__(self):
+            return "Hello"
+
+    fake_convertor = munits.ConversionInterface()
+    # v, u, a = value, unit, axis
+    fake_convertor.convert = Mock(side_effect=lambda v, u, a: 0)
+    # not used, here for completeness
+    fake_convertor.default_units = Mock(side_effect=lambda v, a: None)
+    fake_convertor.axisinfo = Mock(side_effect=lambda u, a: munits.AxisInfo())
+
+    munits.registry[FakeUnit] = fake_convertor
+
+    data = [[FakeUnit("yellow"), FakeUnit(42)],
+            [FakeUnit(datetime.datetime(1968, 8, 1)), FakeUnit(True)]]
+
+    fig_test.subplots().table(data)
+    fig_ref.subplots().table([["Hello", "Hello"], ["Hello", "Hello"]])
+    fig_test.canvas.draw()
+    fake_convertor.convert.assert_not_called()
+
+    munits.registry.pop(FakeUnit)
+    assert not munits.registry.get_converter(FakeUnit)
