Add renko and pnf tests

Author: coffincw

tests/reference_images/pnf01.png

@@ -59,12 +59,16 @@ def test_kwarg_validation_error(bolldata):
         mpf.make_marketcolors(base_mpf_style='classic',ohlc='chartreussse')
     assert 'NOT is_color_like' in str(ex.value)
 
-def test_renko_addplot(bolldata):
+def test_pmove_addplot(bolldata):
     df = bolldata
     apdict = mpf.make_addplot(df['LowerB'])
     with pytest.raises(ValueError) as ex:
         mpf.plot(df,type='renko',volume=True,addplot=apdict)
     assert '`addplot` is not supported for `type=\'renko\'`' in str(ex.value)
+
+    with pytest.raises(ValueError) as ex:
+        mpf.plot(df,type='pointnfigure',volume=True,addplot=apdict)
+    assert '`addplot` is not supported for `type=\'pointnfigure\'`' in str(ex.value)
     #mpf.plot(df,type='renko',volume=True)
 
 def test_figratio_bounds(bolldata):

tests/reference_images/pnf02.png

@@ -0,0 +1,111 @@
+import os
+import os.path
+import glob
+import mplfinance        as mpf
+import matplotlib.pyplot as plt
+from   matplotlib.testing.compare import compare_images
+
+print('mpf.__version__ =',mpf.__version__)                 # for the record
+print("plt.rcParams['backend'] =",plt.rcParams['backend']) # for the record
+
+base='pnf'
+tdir = os.path.join('tests','test_images')
+refd = os.path.join('tests','reference_images')
+
+globpattern = os.path.join(tdir,base+'*.png')
+oldtestfiles = glob.glob(globpattern)
+
+for fn in oldtestfiles:
+    try:
+        os.remove(fn)
+    except:
+        print('Error removing file "'+fn+'"')
+
+# IMGCOMP_TOLERANCE = 7.0  # this works fine for linux
+IMGCOMP_TOLERANCE = 11.0  # required for a windows pass. (really 10.25 may do it).
+
+def test_pnf01(bolldata):
+
+    df = bolldata
+
+    fname = base+'01.png'
+    tname = os.path.join(tdir,fname)
+    rname = os.path.join(refd,fname)
+
+    mpf.plot(df,type='pnf', volume=True,savefig=tname)
+   
+    tsize = os.path.getsize(tname)
+    print(glob.glob(tname),'[',tsize,'bytes',']')
+
+    rsize = os.path.getsize(rname)
+    print(glob.glob(rname),'[',rsize,'bytes',']')
+
+    result = compare_images(rname,tname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None
+
+
+def test_pnf02(bolldata):
+
+    df = bolldata
+
+    fname = base+'02.png'
+    tname = os.path.join(tdir,fname)
+    rname = os.path.join(refd,fname)
+
+    mpf.plot(df,type='pnf', pointnfig_params=dict(box_size=4), volume=True, savefig=tname)
+   
+    tsize = os.path.getsize(tname)
+    print(glob.glob(tname),'[',tsize,'bytes',']')
+
+    rsize = os.path.getsize(rname)
+    print(glob.glob(rname),'[',rsize,'bytes',']')
+
+    result = compare_images(rname,tname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None
+
+
+def test_pnf03(bolldata):
+
+    df = bolldata
+
+    fname = base+'03.png'
+    tname = os.path.join(tdir,fname)
+    rname = os.path.join(refd,fname)
+
+    mpf.plot(df,type='pnf', pointnfig_params=dict(box_size='atr', atr_length=2), volume=True, savefig=tname)
+   
+    tsize = os.path.getsize(tname)
+    print(glob.glob(tname),'[',tsize,'bytes',']')
+
+    rsize = os.path.getsize(rname)
+    print(glob.glob(rname),'[',rsize,'bytes',']')
+
+    result = compare_images(rname,tname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None
+
+def test_pnf04(bolldata):
+
+    df = bolldata
+
+    fname = base+'04.png'
+    tname = os.path.join(tdir,fname)
+    rname = os.path.join(refd,fname)
+
+    mpf.plot(df,type='pnf', pointnfig_params=dict(box_size='atr', atr_length='total'), mav=(4,6,8), volume=True, savefig=tname)
+   
+    tsize = os.path.getsize(tname)
+    print(glob.glob(tname),'[',tsize,'bytes',']')
+
+    rsize = os.path.getsize(rname)
+    print(glob.glob(rname),'[',rsize,'bytes',']')
+
+    result = compare_images(rname,tname,tol=IMGCOMP_TOLERANCE)
+    if result is not None:
+       print('result=',result)
+    assert result is None