Add some tests for cbook._define_aliases.

QuLogic · QuLogic · commit 687906efdaa2 · 2020-09-12T04:13:11.000-04:00
diff --git a/lib/matplotlib/tests/test_cbook.py b/lib/matplotlib/tests/test_cbook.py
@@ -298,6 +298,39 @@ def test_sanitize_sequence():
     assert k == cbook.sanitize_sequence(k)
 
 
+def test_define_aliases():
+    @cbook._define_aliases({'prop': ['alias1', 'alias2']})
+    class NoGetter:
+        def set_prop(self):
+            pass
+
+    @cbook._define_aliases({'prop': ['alias1', 'alias2']})
+    class NoSetter:
+        def get_prop(self):
+            pass
+
+    assert getattr(NoGetter, 'get_alias1', None) is None
+    assert getattr(NoGetter, 'set_alias1', None) is not None
+    assert getattr(NoGetter, 'get_alias2', None) is None
+    assert getattr(NoGetter, 'set_alias2', None) is not None
+    assert NoGetter._alias_map == {'prop': ['alias1', 'alias2']}
+
+    assert getattr(NoSetter, 'get_alias1', None) is not None
+    assert getattr(NoSetter, 'set_alias1', None) is None
+    assert getattr(NoSetter, 'get_alias2', None) is not None
+    assert getattr(NoSetter, 'set_alias2', None) is None
+    assert NoSetter._alias_map == {'prop': ['alias1', 'alias2']}
+
+
+def test_define_aliases_invalid():
+    class Invalid:
+        pass
+
+    with pytest.raises(ValueError,
+                       match="Neither getter nor setter exists for 'prop'"):
+        cbook._define_aliases({'prop': ['alias1', 'alias2']})(Invalid)
+
+
 fail_mapping = (
     ({'a': 1}, {'forbidden': ('a')}),
     ({'a': 1}, {'required': ('b')}),
