more test fixes

Author: nschloe

tests/test_annotate.py

@@ -1,8 +1,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     fig = plt.figure(1, figsize=(8, 5))
@@ -37,11 +35,6 @@ def plot():
 
 
 def test():
-    assert_equality(plot, __file__[:-3] + "_reference.tex")
-
+    from .helpers import assert_equality
 
-if __name__ == "__main__":
-    import helpers
-
-    helpers.compare_mpl_tex(plot)
-    # helpers.print_tree(plot())
+    assert_equality(plot, __file__[:-3] + "_reference.tex")

tests/test_arrows.py

@@ -66,9 +66,3 @@ def to_texstring(s):
 #     code = tikzplotlib.get_tikz_code(include_disclaimer=False, standalone=True)
 #     plt.close()
 #     helpers._does_compile(code)
-
-if __name__ == "__main__":
-    import helpers
-
-    helpers.compare_mpl_tex(plot)
-    # helpers.print_tree(plot())

tests/test_axvline.py

@@ -1,8 +1,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     fig = plt.figure()
@@ -15,4 +13,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, __file__[:-3] + "_reference.tex")

tests/test_barchart.py

@@ -6,8 +6,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     # plot data
@@ -28,4 +26,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_barchart_reference.tex")

tests/test_barchart_errorbars.py

@@ -6,8 +6,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     # plot data
@@ -33,4 +31,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_barchart_errorbars_reference.tex")

tests/test_barchart_legend.py

@@ -10,8 +10,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     # plot data
@@ -33,4 +31,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_barchart_legend_reference.tex")

tests/test_basic_sin.py

@@ -1,8 +1,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from .helpers import assert_equality
-
 
 def plot():
     fig = plt.figure()
@@ -23,4 +21,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_basic_sin_reference.tex")

tests/test_boxplot.py

@@ -7,8 +7,6 @@
 """
 import matplotlib.pyplot as plt
 
-from .helpers import assert_equality
-
 
 def plot():
     # plot data
@@ -90,4 +88,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_boxplot_reference.tex")

tests/test_colorbars.py

@@ -1,8 +1,6 @@
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
-from .helpers import assert_equality
-
 
 def plot():
     # Make a figure and axes with dimensions as desired.
@@ -75,4 +73,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, "test_colorbars_reference.tex", assert_compilation=False)

tests/test_context.py

@@ -1,6 +1,3 @@
-from .helpers import assert_equality
-
-
 def plot():
     import matplotlib.cm as cm
     import matplotlib.pyplot as plt
@@ -16,4 +13,6 @@ def plot():
 
 
 def test():
+    from .helpers import assert_equality
+
     assert_equality(plot, __file__[:-3] + "_reference.tex", flavor="context")