Add unshare tests.

lkjell · lkjell · commit 87fee275f194 · 2018-03-09T11:28:55.000+01:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -5737,3 +5737,125 @@ def test_share_symmetric1():
         assert _symmetric(A, B, ax)
 
         plt.close(fig)
+
+
+def test_unshare_axes():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(511)
+        B = fig.add_subplot(512)
+        C = fig.add_subplot(513)
+        D = fig.add_subplot(514)
+        E = fig.add_subplot(515)
+
+        # E -> A
+        share = getattr(E, "share_{}_axes".format(ax))
+        share(A, symmetric=False, transitive=False)
+
+        # C -> D
+        share = getattr(C, "share_{}_axes".format(ax))
+        share(D, symmetric=False, transitive=False)
+
+        # A -> B
+        share = getattr(A, "share_{}_axes".format(ax))
+        share(B, symmetric=False, transitive=False)
+
+        # A -> C, A -> D
+        share(C, symmetric=False, transitive=True)
+
+        unshare = getattr(A, "unshare_{}_axes".format(ax))
+        unshare(None)
+
+        assert _asymmetric(C, D, ax)
+        assert _no_relation(A, B, ax)
+        assert _no_relation(A, C, ax)
+        assert _no_relation(A, D, ax)
+        assert _no_relation(A, E, ax)
+
+        assert _no_relation(B, C, ax)
+        assert _no_relation(B, E, ax)
+        assert _no_relation(B, D, ax)
+
+        assert _no_relation(E, B, ax)
+        assert _no_relation(E, C, ax)
+        assert _no_relation(E, D, ax)
+
+
+def test_unshare_axes1():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(511)
+        B = fig.add_subplot(512)
+        C = fig.add_subplot(513)
+        D = fig.add_subplot(514)
+        E = fig.add_subplot(515)
+
+        # E -> A
+        share = getattr(E, "share_{}_axes".format(ax))
+        share(A, symmetric=False, transitive=False)
+
+        # C -> D
+        share = getattr(C, "share_{}_axes".format(ax))
+        share(D, symmetric=False, transitive=False)
+
+        # A -> B
+        share = getattr(A, "share_{}_axes".format(ax))
+        share(B, symmetric=False, transitive=False)
+
+        # A -> C, A -> D
+        share(C, symmetric=False, transitive=True)
+
+        unshare = getattr(A, "unshare_{}_axes".format(ax))
+        unshare(A)
+
+        assert _asymmetric(C, D, ax)
+        assert _asymmetric(A, B, ax)
+        assert _asymmetric(A, C, ax)
+        assert _asymmetric(A, D, ax)
+        assert _no_relation(A, E, ax)
+        assert _no_relation(E, C, ax)
+        assert _no_relation(E, D, ax)
+        assert _no_relation(B, D, ax)
+
+
+def test_unshare_axes2():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(511)
+        B = fig.add_subplot(512)
+        C = fig.add_subplot(513)
+        D = fig.add_subplot(514)
+        E = fig.add_subplot(515)
+
+        # E -> A
+        share = getattr(E, "share_{}_axes".format(ax))
+        share(A, symmetric=False, transitive=False)
+
+        # C -> D
+        share = getattr(C, "share_{}_axes".format(ax))
+        share(D, symmetric=False, transitive=False)
+
+        # A -> B
+        share = getattr(A, "share_{}_axes".format(ax))
+        share(B, symmetric=False, transitive=False)
+
+        # A -> C, A -> D
+        share(C, symmetric=False, transitive=True)
+
+        unshare = getattr(A, "unshare_{}_axes".format(ax))
+        unshare(C)
+
+        assert _asymmetric(C, D, ax)
+        assert _asymmetric(A, B, ax)
+        assert _no_relation(A, C, ax)
+        assert _asymmetric(A, D, ax)
+        assert _asymmetric(E, A, ax)
+        assert _no_relation(E, C, ax)
+        assert _no_relation(E, D, ax)
+        assert _no_relation(B, D, ax)
