Merge pull request #98 from coffincw/master

Add Renko and Point and Figure Tests

Author: DanielGoldfarb

tests/reference_images/pnf01.png

@@ -32,8 +32,9 @@ def test_addplot01(bolldata):
     tname = os.path.join(tdir,fname)
     rname = os.path.join(refd,fname)
 
-    mpf.plot(df,volume=True,savefig=tname)
-   
+    fig_axis = mpf.plot(df,volume=True,savefig=tname,returnfig=True)
+    plt.close(fig_axis[0])
+
     tsize = os.path.getsize(tname)
     print(glob.glob(tname),'[',tsize,'bytes',']')
 
@@ -53,7 +54,8 @@ def test_addplot02(bolldata):
     rname = os.path.join(refd,fname)
 
     apdict = mpf.make_addplot(df['LowerB'])
-    mpf.plot(df,volume=True,addplot=apdict,savefig=tname)
+    fig_axis = mpf.plot(df,volume=True,addplot=apdict,savefig=tname,returnfig=True)
+    plt.close(fig_axis[0])
 
     tsize = os.path.getsize(tname)
     print(glob.glob(tname),'[',tsize,'bytes',']')
@@ -108,7 +110,8 @@ def test_addplot03(bolldata):
              mpf.make_addplot((df['PercentB']),panel='lower',color='g')
            ]
 
-    mpf.plot(df,addplot=apds,figscale=1.3,volume=True,savefig=tname)
+    fig_axis = mpf.plot(df,addplot=apds,figscale=1.3,volume=True,savefig=tname,returnfig=True)
+    plt.close(fig_axis[0])
 
     tsize = os.path.getsize(tname)
     print(glob.glob(tname),'[',tsize,'bytes',']')
@@ -139,8 +142,9 @@ def test_addplot04(bolldata):
              mpf.make_addplot((df['PercentB']),panel='lower',color='g',linestyle='dotted')
            ]
 
-    mpf.plot(df,addplot=apds,figscale=1.5,volume=True,
-             style='starsandstripes',savefig=tname)
+    fig_axis = mpf.plot(df,addplot=apds,figscale=1.5,volume=True,
+             style='starsandstripes',savefig=tname,returnfig=True)
+    plt.close(fig_axis[0])
 
     tsize = os.path.getsize(tname)
     print(glob.glob(tname),'[',tsize,'bytes',']')

tests/reference_images/pnf02.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):