Skip to content

Commit 73d8920

Browse files
committed
update tests
1 parent 52b7bd9 commit 73d8920

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

63 files changed

+204
-308
lines changed

tests/test_annotate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import matplotlib.pyplot as plt
22
import numpy as np
3-
from helpers import assert_equality
3+
4+
from .helpers import assert_equality
45

56

67
def plot():

tests/test_axvline.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import matplotlib.pyplot as plt
22
import numpy as np
3-
from helpers import assert_equality
3+
4+
from .helpers import assert_equality
45

56

67
def plot():
@@ -15,11 +16,3 @@ def plot():
1516

1617
def test():
1718
assert_equality(plot, __file__[:-3] + "_reference.tex")
18-
return
19-
20-
21-
if __name__ == "__main__":
22-
import helpers
23-
24-
helpers.compare_mpl_tex(plot)
25-
# helpers.print_tree(plot())

tests/test_barchart.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
"""
66
import matplotlib.pyplot as plt
77
import numpy as np
8-
from helpers import assert_equality
8+
9+
from .helpers import assert_equality
910

1011

1112
def plot():
@@ -28,4 +29,3 @@ def plot():
2829

2930
def test():
3031
assert_equality(plot, "test_barchart_reference.tex")
31-
return

tests/test_barchart_errorbars.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
"""
66
import matplotlib.pyplot as plt
77
import numpy as np
8-
from helpers import assert_equality
8+
9+
from .helpers import assert_equality
910

1011

1112
def plot():
@@ -33,4 +34,3 @@ def plot():
3334

3435
def test():
3536
assert_equality(plot, "test_barchart_errorbars_reference.tex")
36-
return

tests/test_barchart_legend.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99
"""
1010
import matplotlib.pyplot as plt
1111
import numpy as np
12-
from helpers import assert_equality
12+
13+
from .helpers import assert_equality
1314

1415

1516
def plot():
@@ -33,10 +34,3 @@ def plot():
3334

3435
def test():
3536
assert_equality(plot, "test_barchart_legend_reference.tex")
36-
37-
38-
if __name__ == "__main__":
39-
import helpers
40-
41-
helpers.compare_mpl_tex(plot)
42-
# helpers.print_tree(plot())

tests/test_basic_sin.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import matplotlib.pyplot as plt
22
import numpy as np
3-
from helpers import assert_equality
3+
4+
from .helpers import assert_equality
45

56

67
def plot():
@@ -23,10 +24,3 @@ def plot():
2324

2425
def test():
2526
assert_equality(plot, "test_basic_sin_reference.tex")
26-
27-
28-
if __name__ == "__main__":
29-
import helpers
30-
31-
helpers.compare_mpl_tex(plot)
32-
# helpers.print_tree(plot())

tests/test_boxplot.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
See: https://github.com/nschloe/tikzplotlib/pull/134
77
"""
88
import matplotlib.pyplot as plt
9-
from helpers import assert_equality
9+
10+
from .helpers import assert_equality
1011

1112

1213
def plot():
@@ -90,8 +91,3 @@ def plot():
9091

9192
def test():
9293
assert_equality(plot, "test_boxplot_reference.tex")
93-
94-
95-
if __name__ == "__main__":
96-
plot()
97-
plt.show()

tests/test_colorbars.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import matplotlib as mpl
22
import matplotlib.pyplot as plt
3-
from helpers import assert_equality
3+
4+
from .helpers import assert_equality
45

56

67
def plot():
@@ -75,10 +76,3 @@ def plot():
7576

7677
def test():
7778
assert_equality(plot, "test_colorbars_reference.tex", assert_compilation=False)
78-
79-
80-
if __name__ == "__main__":
81-
import helpers
82-
83-
helpers.compare_mpl_tex(plot)
84-
# helpers.print_tree(plot())

tests/test_context.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from helpers import assert_equality
1+
from .helpers import assert_equality
22

33

44
def plot():
@@ -17,4 +17,3 @@ def plot():
1717

1818
def test():
1919
assert_equality(plot, __file__[:-3] + "_reference.tex", flavor="context")
20-
return

tests/test_contourf.py

Lines changed: 2 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import matplotlib.pyplot as plt
22
import numpy as np
3-
from helpers import assert_equality
3+
4+
from .helpers import assert_equality
45

56

67
def plot():
@@ -28,17 +29,3 @@ def plot():
2829

2930
def test():
3031
assert_equality(plot, __file__[:-3] + "_reference.tex")
31-
32-
33-
if __name__ == "__main__":
34-
# plot()
35-
# plt.show()
36-
# import helpers
37-
# helpers.compare_mpl_tex(plot)
38-
# helpers.print_tree(plot())
39-
plot()
40-
# plt.show()
41-
# plt.savefig('out.pgf')
42-
import tikzplotlib
43-
44-
tikzplotlib.save("out.tex", standalone=True)

0 commit comments

Comments
 (0)