restructure tests

Author: DanielGoldfarb

examples/mpftests/test_addplot.py

@@ -0,0 +1,54 @@
+import os                as os
+import pandas            as pd
+import mplfinance        as mpf
+import matplotlib.pyplot as plt
+from   matplotlib.testing.compare import compare_images
+
+print('pd.__version__  =',pd.__version__ )                 # for the record
+print('mpf.__version__ =',mpf.__version__)                 # for the record
+print("plt.rcParams['backend'] =",plt.rcParams['backend']) # for the record
+
+import subprocess
+pwd = subprocess.run(['pwd'], stdout=subprocess.PIPE)
+print('pwd.stdout=',pwd.stdout)
+
+df = pd.read_csv('examples/data/SPY_20110701_20120630_Bollinger.csv',index_col=0,parse_dates=True)
+print('df.shape='  , df.shape  )
+print('df.head(3)=', df.head(3))
+print('df.tail(3)=', df.tail(3))
+
+prefix='addplot'
+tdir='tests/mpftests/test_images/'
+refd='tests/mpftests/reference_images/'
+#os.system('rm -f '+tdir+prefix+'*.jpg')
+os.system('rm -f '+tdir+prefix+'*.png')
+
+IMGCOMP_TOLERANCE = 7.0
+
+# ---- Test 01 -----
+
+def test_addplot01():
+
+    fname=prefix+'01.png'
+    mpf.plot(df,volume=True,savefig=tdir+fname)
+
+    os.system('ls -l '+tdir+fname)
+
+    result = compare_images(refd+fname,tdir+fname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None
+
+# ---- Test 02 -----
+
+def test_addplot02():
+    fname=prefix+'02.png'
+    apdict = mpf.make_addplot(df['LowerB'])
+    mpf.plot(df,volume=True,addplot=apdict,savefig=tdir+fname)
+
+    os.system('ls -l '+tdir+fname)
+
+    result = compare_images(refd+fname,tdir+fname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None

examples/mpftests/reference_images/addplot01.png renamed to tests/mpftests/reference_images/addplot01.png

@@ -0,0 +1,37 @@
+import os                as os
+import pandas            as pd
+import mplfinance        as mpf
+import matplotlib.pyplot as plt
+from   matplotlib.testing.compare import compare_images
+
+print('pd.__version__  =',pd.__version__ )                 # for the record
+print('mpf.__version__ =',mpf.__version__)                 # for the record
+print("plt.rcParams['backend'] =",plt.rcParams['backend']) # for the record
+
+import subprocess
+pwd = subprocess.run(['pwd'], stdout=subprocess.PIPE)
+print('pwd.stdout=',pwd.stdout)
+
+df = pd.read_csv('examples/data/SPY_20110701_20120630_Bollinger.csv',index_col=0,parse_dates=True)
+print('df.shape='  , df.shape  )
+print('df.head(3)=', df.head(3))
+print('df.tail(3)=', df.tail(3))
+
+##  prefix='addplot'
+##  tdir='test_images/'
+##  refd='reference_images/'
+##  #os.system('rm -f '+tdir+prefix+'*.jpg')
+##  os.system('rm -f '+tdir+prefix+'*.png')
+
+import pytest
+
+#   @pytest.fixture(scope='session',autouse=True)
+#   def root_directory(request):
+#       return str(request.config.rootdir)
+
+#def test_raise_ex(root_directory):
+def test_raise():
+    #print("root_directory=",root_directory)
+    apdict = mpf.make_addplot(df['LowerB'])
+    with pytest.raises(ValueError):
+        mpf.plot(df,volume='True',addplot=apdict)

examples/mpftests/reference_images/addplot02.png renamed to tests/mpftests/reference_images/addplot02.png

@@ -2,29 +2,15 @@
 import os
 import subprocess
 
-def test_mplfinance():
-    '''Run new mplfinance api tests.'''
-
-    os.environ['MPLBACKEND'] = 'agg'
-
-    os.chdir("examples/mpftests")
-
-    exs = glob.glob('test*.py')
-    try:
-        for ex in exs:
-            subprocess.check_call(["python", ex])
-    finally:
-        os.chdir("../..")
-
-
 def test_original_flavor_e2e_examples():
     """Run e2e tests for  folder"""
 
+    os.environ['MPLBACKEND'] = 'agg'
     os.chdir("examples/original_flavor")
 
     exs = glob.glob('*.py')
     try:
         for ex in exs:
             subprocess.check_call(["python", ex])
     finally:
-        os.chdir("../..")
+        os.chdir("../../")

tests/mpftests/test_addplot.py

@@ -6,8 +6,8 @@
 [tox]
 envlist = py36, py37, py38
 
-[pytest]
-python_files = tests.py
+#[pytest]
+#python_files = tests.py
 
 [testenv]
 deps =