update some tests

Author: nschloe

tests/test_custom_collection.py

@@ -6,7 +6,8 @@
 import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np
-from helpers import assert_equality
+
+from .helpers import assert_equality
 
 
 class TransformedEllipseCollection(matplotlib.collections.Collection):
@@ -53,7 +54,7 @@ def plot():
     fig = plt.figure()
     ax = fig.add_subplot(111)
 
-    theta = np.linspace(0, 2 * np.pi, 12, endpoint=False)
+    theta = np.linspace(0, 2 * np.pi, 6, endpoint=False)
     mats = rot(theta) @ np.diag([0.1, 0.2])
     x = np.cos(theta)
     y = np.sin(theta)
@@ -74,4 +75,3 @@ def plot():
 
 def test():
     assert_equality(plot, "test_custom_collection_reference.tex")
-    return

tests/test_subplot4x4.py

@@ -1,39 +1,38 @@
-from helpers import assert_equality
+from .helpers import assert_equality
 
 
 def plot():
     import numpy as np
-    from matplotlib import pyplot as pp
+    from matplotlib import pyplot as plt
 
-    fig = pp.figure()
+    fig = plt.figure()
 
     an = np.linspace(0, 2 * np.pi, 10)
 
-    pp.subplot(221)
-    pp.plot(3 * np.cos(an), 3 * np.sin(an))
-    pp.title("not equal, looks like ellipse", fontsize=10)
+    plt.subplot(221)
+    plt.plot(3 * np.cos(an), 3 * np.sin(an))
+    plt.title("not equal, looks like ellipse", fontsize=10)
 
-    pp.subplot(222)
-    pp.plot(3 * np.cos(an), 3 * np.sin(an))
-    pp.axis("equal")
-    pp.title("equal, looks like circle", fontsize=10)
+    plt.subplot(222)
+    plt.plot(3 * np.cos(an), 3 * np.sin(an))
+    plt.axis("equal")
+    plt.title("equal, looks like circle", fontsize=10)
 
-    pp.subplot(223)
-    pp.plot(3 * np.cos(an), 3 * np.sin(an))
-    pp.axis("equal")
-    pp.axis([-3, 3, -3, 3])
-    pp.title("looks like circle, even after changing limits", fontsize=10)
+    plt.subplot(223)
+    plt.plot(3 * np.cos(an), 3 * np.sin(an))
+    plt.axis("equal")
+    plt.axis([-3, 3, -3, 3])
+    plt.title("looks like circle, even after changing limits", fontsize=10)
 
-    pp.subplot(224)
-    pp.plot(3 * np.cos(an), 3 * np.sin(an))
-    pp.axis("equal")
-    pp.axis([-3, 3, -3, 3])
-    pp.plot([0, 4], [0, 4])
-    pp.title("still equal after adding line", fontsize=10)
+    plt.subplot(224)
+    plt.plot(3 * np.cos(an), 3 * np.sin(an))
+    plt.axis("equal")
+    plt.axis([-3, 3, -3, 3])
+    plt.plot([0, 4], [0, 4])
+    plt.title("still equal after adding line", fontsize=10)
 
     return fig
 
 
 def test():
     assert_equality(plot, __file__[:-3] + "_reference.tex")
-    return