def test_plot_oth(self): #just test that they run res = self.res endog = res.model.endog exog = res.model.exog plot_fit(res, 0, y_true=None) plot_partregress(endog, exog, exog_idx=[0,1]) plot_regress_exog(res, exog_idx=[0]) plot_ccpr(res, exog_idx=[0]) plot_ccpr(res, exog_idx=[0,1]) plt.close('all')
def test_plot_oth(self): #just test that they run res = self.res endog = res.model.endog exog = res.model.exog plot_fit(res, 0, y_true=None) plot_partregress(endog, exog, exog_idx=[0, 1]) plot_regress_exog(res, exog_idx=[0]) plot_ccpr(res, exog_idx=[0]) plot_ccpr(res, exog_idx=[0, 1]) plt.close('all')
def test_plot_fit(self): res = self.res fig = plot_fit(res, 0, y_true=None) x0 = res.model.exog[:, 0] yf = res.fittedvalues y = res.model.endog px1, px2 = fig.axes[0].get_lines()[0].get_data() np.testing.assert_equal(x0, px1) np.testing.assert_equal(y, px2) px1, px2 = fig.axes[0].get_lines()[1].get_data() np.testing.assert_equal(x0, px1) np.testing.assert_equal(yf, px2) plt.close(fig)