Adding symmetric and transitive share test.

lkjell · lkjell · commit 3cc9213ab957 · 2018-03-09T10:49:56.000+01:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -5601,69 +5601,139 @@ def test_plot_columns_cycle_deprecation():
         plt.plot(np.zeros((2, 2)), np.zeros((2, 3)))
 
 
-# def test_share_symmetric():
-#     fig = plt.figure()
-#     ax1 = fig.add_subplot(211)
-#     ax2 = fig.add_subplot(212)
-#
-#     ax1.share_x_axes(ax2, symmetric=True, transitive=False)
-
-#
-# def test_share_unshare_3d_axes():
-#     fig = plt.figure()
-#     ax1 = fig.add_subplot(211, projection="3d")
-#     ax2 = fig.add_subplot(212, projection="3d",
-#                           sharex=ax1,
-#                           sharey=ax1,
-#                           sharez=ax1)
-#
-#     # Testing unsharing
-#     ax1.unshare_axes()
-#     assert ax2._sharex is None
-#     assert ax2._sharey is None
-#     assert ax2._sharez is None
-#
-#     sharedx = ax2._shared_x_axes
-#     sharedy = ax2._shared_y_axes
-#     sharedz = ax2._shared_z_axes
-#
-#     assert ax1 not in sharedx
-#     assert ax1 not in sharedy
-#     assert ax1 not in sharedz
-#
-#     sharedx = ax1._shared_x_axes
-#     sharedy = ax1._shared_y_axes
-#     sharedz = ax1._shared_z_axes
-#
-#     assert ax2 not in sharedx
-#     assert ax2 not in sharedy
-#     assert ax2 not in sharedz
-#
-#     # Testing sharing
-#
-#     # x axis
-#     ax1.share_x_axes(ax2)
-#
-#     sharedx = ax1._shared_x_axes
-#     assert ax2 in sharedx
-#
-#     sharedx = ax2._shared_x_axes
-#     assert ax1 in sharedx
-#
-#     # y axis
-#     ax2.share_y_axes(ax1)
-#
-#     sharedy = ax1._shared_y_axes
-#     assert ax2 in sharedy
-#
-#     sharedy = ax2._shared_y_axes
-#     assert ax1 in sharedy
-#
-#     # z axis
-#     ax1.share_z_axes(ax2)
-#
-#     sharedz = ax2._shared_z_axes
-#     assert ax1 in sharedz
-#
-#     sharedz = ax1._shared_z_axes
-#     assert ax2 in sharedz
+def _symmetric(A, B, shared_axes):
+    # Symmetric test between A and C
+    C = B in getattr(A, "get_shared_{}_axes".format(shared_axes))()
+    C &= A in getattr(B, "get_shared_{}_axes".format(shared_axes))()
+    C &= B in getattr(A, "get_shared_{}_inv_axes".format(shared_axes))()
+    C &= A in getattr(B, "get_shared_{}_inv_axes".format(shared_axes))()
+    return C
+
+
+def _no_relation(A, B, shared_axes):
+    # Symmetric test between A and C
+    C = B not in getattr(A, "get_shared_{}_axes".format(shared_axes))()
+    C &= A not in getattr(B, "get_shared_{}_axes".format(shared_axes))()
+    C &= B not in getattr(A, "get_shared_{}_inv_axes".format(shared_axes))()
+    C &= A not in getattr(B, "get_shared_{}_inv_axes".format(shared_axes))()
+    return C
+
+
+def _asymmetric(A, B, shared_axes):
+    # A -> B but not B -> A
+    C = B in getattr(A, "get_shared_{}_axes".format(shared_axes))()
+    C &= A not in getattr(B, "get_shared_{}_axes".format(shared_axes))()
+    C &= A in getattr(B, "get_shared_{}_inv_axes".format(shared_axes))()
+    C &= B not in getattr(A, "get_shared_{}_inv_axes".format(shared_axes))()
+    return C
+
+
+def test_share_transitive():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(411)
+        B = fig.add_subplot(412)
+        C = fig.add_subplot(413)
+        D = fig.add_subplot(414)
+
+        # 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, C -> A, C -> B, A -> D
+        share(C, symmetric=True, transitive=True)
+
+        # Symmetric test between A and C
+        assert _symmetric(A, C, ax)
+
+        # A -> B but not B -> C
+        assert _asymmetric(A, B, ax)
+
+        # C -> B but not B -> C
+        assert _asymmetric(C, B, ax)
+
+        # A -> D but not D -> D
+        assert _asymmetric(A, D, ax)
+
+        # C -> D but not D -> C
+        assert _asymmetric(C, D, ax)
+
+        assert _no_relation(B, D, ax)
+
+        plt.close(fig)
+
+
+def test_share_transitive1():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(411)
+        B = fig.add_subplot(412)
+        C = fig.add_subplot(413)
+        D = fig.add_subplot(414)
+
+        # 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)
+
+        # A -> C but not C -> A
+        assert _asymmetric(A, C, ax)
+
+        # A -> B but not B -> C
+        assert _asymmetric(A, B, ax)
+
+        # A -> D but not D -> A
+        assert _asymmetric(A, D, ax)
+
+        # C -> D but not D -> C
+        assert _asymmetric(C, D, ax)
+
+        assert _no_relation(B, D, ax)
+        assert _no_relation(B, C, ax)
+
+        plt.close(fig)
+
+
+def test_share_symmetric():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(211)
+        B = fig.add_subplot(212)
+
+        share = getattr(A, "share_{}_axes".format(ax))
+        # A -> B but not B -> A
+        share(B, symmetric=False, transitive=False)
+        assert _asymmetric(A, B, ax)
+        plt.close(fig)
+
+
+def test_share_symmetric1():
+    axes = ["x", "y"]
+
+    for ax in axes:
+        fig = plt.figure()
+        A = fig.add_subplot(211)
+        B = fig.add_subplot(212)
+
+        share = getattr(A, "share_{}_axes".format(ax))
+        # A -> B, B -> A
+        share(B, symmetric=True, transitive=False)
+        assert _symmetric(A, B, ax)
+
+        plt.close(fig)
