further re-org tests

Author: DanielGoldfarb

tests/conftest.py

@@ -0,0 +1,17 @@
+import pytest
+import os                as os
+import pandas            as pd
+
+os.environ['MPLBACKEND'] = 'agg'
+
+print('\npd.__version__  =',pd.__version__ )                 # for the record
+
+
+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))
+
+@pytest.fixture()
+def bolldata():
+   return df

tests/mpftests/test_raise.py

@@ -1,33 +1,25 @@
 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))
+print('pwd.stdout=',str(pwd.stdout).strip())
 
 prefix='addplot'
-tdir='tests/mpftests/test_images/'
-refd='tests/mpftests/reference_images/'
-#os.system('rm -f '+tdir+prefix+'*.jpg')
+tdir='tests/test_images/'
+refd='tests/reference_images/'
 os.system('rm -f '+tdir+prefix+'*.png')
 
 IMGCOMP_TOLERANCE = 7.0
 
-# ---- Test 01 -----
+def test_addplot01(bolldata):
 
-def test_addplot01():
+    df = bolldata
 
     fname=prefix+'01.png'
     mpf.plot(df,volume=True,savefig=tdir+fname)
@@ -39,9 +31,8 @@ def test_addplot01():
        print('result=',result)
     assert result is None
 
-# ---- Test 02 -----
-
-def test_addplot02():
+def test_addplot02(bolldata):
+    df = bolldata
     fname=prefix+'02.png'
     apdict = mpf.make_addplot(df['LowerB'])
     mpf.plot(df,volume=True,addplot=apdict,savefig=tdir+fname)

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

@@ -5,9 +5,7 @@
 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:

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

@@ -0,0 +1,17 @@
+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 pytest
+
+def test_raise(bolldata):
+    df = bolldata
+    apdict = mpf.make_addplot(df['LowerB'])
+    with pytest.raises(ValueError):
+        mpf.plot(df,volume='True',addplot=apdict)