9
9
from .helper import with_setup
10
10
11
11
TESTPATH = dirname (__file__ )
12
- GENPATH = join (TESTPATH , "dotexport" )
13
12
REFPATH = join (TESTPATH , "refdata" )
14
13
15
14
16
- def setup ():
17
- if not exists (GENPATH ):
18
- makedirs (GENPATH )
19
-
20
-
21
- def teardown ():
22
- if exists (GENPATH ):
23
- rmtree (GENPATH )
24
-
25
-
26
- @with_setup (setup , teardown )
27
- def test_tree1 ():
15
+ def test_tree1 (tmpdir ):
28
16
"""Tree1."""
29
17
root = Node ("root" )
30
18
s0 = Node ("sub0" , parent = root )
@@ -36,12 +24,11 @@ def test_tree1():
36
24
s1c = Node ("sub1C" , parent = s1 )
37
25
Node (99 , parent = s1c )
38
26
39
- RenderTreeGraph (root ).to_dotfile (join (GENPATH , "tree1.dot" ))
40
- assert cmp (join (GENPATH , "tree1.dot" ), join (REFPATH , "tree1.dot" ))
27
+ RenderTreeGraph (root ).to_dotfile (join (tmpdir , "tree1.dot" ))
28
+ assert cmp (join (tmpdir , "tree1.dot" ), join (REFPATH , "tree1.dot" ))
41
29
42
30
43
- @with_setup (setup , teardown )
44
- def test_tree2 ():
31
+ def test_tree2 (tmpdir ):
45
32
"""Tree2."""
46
33
root = Node ("root" )
47
34
s0 = Node ("sub0" , parent = root , edge = 2 )
@@ -67,12 +54,11 @@ def edgeattrfunc(node, child):
67
54
edgeattrfunc = edgeattrfunc ,
68
55
)
69
56
70
- r .to_dotfile (join (GENPATH , "tree2.dot" ))
71
- assert cmp (join (GENPATH , "tree2.dot" ), join (REFPATH , "tree2.dot" ))
57
+ r .to_dotfile (join (tmpdir , "tree2.dot" ))
58
+ assert cmp (join (tmpdir , "tree2.dot" ), join (REFPATH , "tree2.dot" ))
72
59
73
60
74
- @with_setup (setup , teardown )
75
- def test_tree_png ():
61
+ def test_tree_png (tmpdir ):
76
62
"""Tree to png."""
77
63
root = Node ("root" )
78
64
s0 = Node ("sub0" , parent = root )
@@ -84,4 +70,4 @@ def test_tree_png():
84
70
s1c = Node ("sub1C" , parent = s1 )
85
71
Node ("sub1Ca" , parent = s1c )
86
72
87
- RenderTreeGraph (root ).to_picture (join (GENPATH , "tree1.png" ))
73
+ RenderTreeGraph (root ).to_picture (join (tmpdir , "tree1.png" ))
0 commit comments