1
- import os as os
1
+ import os
2
+ import os .path
3
+ import glob
2
4
import mplfinance as mpf
3
5
import matplotlib .pyplot as plt
4
6
from matplotlib .testing .compare import compare_images
5
7
6
8
print ('mpf.__version__ =' ,mpf .__version__ ) # for the record
7
9
print ("plt.rcParams['backend'] =" ,plt .rcParams ['backend' ]) # for the record
8
10
9
- prefix = 'addplot'
10
- tdir = 'tests/test_images/'
11
- refd = 'tests/reference_images/'
12
- os .system ('rm -f ' + tdir + prefix + '*.png' )
11
+ base = 'addplot'
12
+ tdir = os .path .join ('tests' ,'test_images' )
13
+ refd = os .path .join ('tests' ,'reference_images' )
14
+
15
+ globpattern = os .path .join (tdir ,base + '*.png' )
16
+ oldtestfiles = glob .glob (globpattern )
17
+
18
+ for fn in oldtestfiles :
19
+ try :
20
+ os .remove (fn )
21
+ except :
22
+ print ('Error removing file "' + fn + '"' )
13
23
14
24
IMGCOMP_TOLERANCE = 7.0
15
25
16
26
def test_addplot01 (bolldata ):
17
27
18
28
df = bolldata
19
29
20
- fname = prefix + '01.png'
21
- mpf .plot (df ,volume = True ,savefig = tdir + fname )
30
+ fname = base + '01.png'
31
+ tname = os .path .join (tdir ,fname )
32
+ rname = os .path .join (refd ,fname )
33
+
34
+ mpf .plot (df ,volume = True ,savefig = tname )
35
+
36
+ tsize = os .path .getsize (tname )
37
+ print (glob .glob (tname ),'[' ,tsize ,'bytes' ,']' )
22
38
23
- os .system ('ls -l ' + tdir + fname )
39
+ rsize = os .path .getsize (rname )
40
+ print (glob .glob (rname ),'[' ,rsize ,'bytes' ,']' )
24
41
25
- result = compare_images (refd + fname , tdir + fname ,tol = IMGCOMP_TOLERANCE )
42
+ result = compare_images (rname , tname ,tol = IMGCOMP_TOLERANCE )
26
43
if result is not None :
27
44
print ('result=' ,result )
28
45
assert result is None
29
46
30
47
def test_addplot02 (bolldata ):
31
48
df = bolldata
32
- fname = prefix + '02.png'
49
+
50
+ fname = base + '02.png'
51
+ tname = os .path .join (tdir ,fname )
52
+ rname = os .path .join (refd ,fname )
53
+
33
54
apdict = mpf .make_addplot (df ['LowerB' ])
34
- mpf .plot (df ,volume = True ,addplot = apdict ,savefig = tdir + fname )
55
+ mpf .plot (df ,volume = True ,addplot = apdict ,savefig = tname )
56
+
57
+ tsize = os .path .getsize (tname )
58
+ print (glob .glob (tname ),'[' ,tsize ,'bytes' ,']' )
35
59
36
- os .system ('ls -l ' + tdir + fname )
60
+ rsize = os .path .getsize (rname )
61
+ print (glob .glob (rname ),'[' ,rsize ,'bytes' ,']' )
37
62
38
- result = compare_images (refd + fname , tdir + fname ,tol = IMGCOMP_TOLERANCE )
63
+ result = compare_images (rname , tname ,tol = IMGCOMP_TOLERANCE )
39
64
if result is not None :
40
65
print ('result=' ,result )
66
+ assert result is None
41
67
42
68
def percentB_aboveone (percentB ,price ):
43
69
import numpy as np
@@ -65,7 +91,10 @@ def percentB_belowzero(percentB,price):
65
91
66
92
def test_addplot03 (bolldata ):
67
93
df = bolldata
68
- fname = prefix + '03.png'
94
+
95
+ fname = base + '03.png'
96
+ tname = os .path .join (tdir ,fname )
97
+ rname = os .path .join (refd ,fname )
69
98
70
99
tcdf = df [['LowerB' ,'UpperB' ]] # DataFrame with two columns
71
100
@@ -78,18 +107,25 @@ def test_addplot03(bolldata):
78
107
mpf .make_addplot ((df ['PercentB' ]),panel = 'lower' ,color = 'g' )
79
108
]
80
109
81
- mpf .plot (df ,addplot = apds ,figscale = 1.3 ,volume = True ,savefig = tdir + fname )
110
+ mpf .plot (df ,addplot = apds ,figscale = 1.3 ,volume = True ,savefig = tname )
111
+
112
+ tsize = os .path .getsize (tname )
113
+ print (glob .glob (tname ),'[' ,tsize ,'bytes' ,']' )
82
114
83
- os .system ('ls -l ' + tdir + fname )
115
+ rsize = os .path .getsize (rname )
116
+ print (glob .glob (rname ),'[' ,rsize ,'bytes' ,']' )
84
117
85
- result = compare_images (refd + fname , tdir + fname ,tol = IMGCOMP_TOLERANCE )
118
+ result = compare_images (rname , tname ,tol = IMGCOMP_TOLERANCE )
86
119
if result is not None :
87
120
print ('result=' ,result )
88
121
assert result is None
89
122
90
123
def test_addplot04 (bolldata ):
91
124
df = bolldata
92
- fname = prefix + '04.png'
125
+
126
+ fname = base + '04.png'
127
+ tname = os .path .join (tdir ,fname )
128
+ rname = os .path .join (refd ,fname )
93
129
94
130
tcdf = df [['LowerB' ,'UpperB' ]] # DataFrame with two columns
95
131
@@ -103,11 +139,15 @@ def test_addplot04(bolldata):
103
139
]
104
140
105
141
mpf .plot (df ,addplot = apds ,figscale = 1.5 ,volume = True ,
106
- style = 'starsandstripes' ,savefig = tdir + fname )
142
+ style = 'starsandstripes' ,savefig = tname )
143
+
144
+ tsize = os .path .getsize (tname )
145
+ print (glob .glob (tname ),'[' ,tsize ,'bytes' ,']' )
107
146
108
- os .system ('ls -l ' + tdir + fname )
147
+ rsize = os .path .getsize (rname )
148
+ print (glob .glob (rname ),'[' ,rsize ,'bytes' ,']' )
109
149
110
- result = compare_images (refd + fname , tdir + fname ,tol = IMGCOMP_TOLERANCE )
150
+ result = compare_images (rname , tname ,tol = IMGCOMP_TOLERANCE )
111
151
if result is not None :
112
152
print ('result=' ,result )
113
153
assert result is None
0 commit comments