from statsmodels.compat.testing import skipif import numpy as np import statsmodels.api as sm from statsmodels.graphics.gofplots import qqplot, qqline, ProbPlot from scipy import stats try: import matplotlib.pyplot as plt import matplotlib have_matplotlib = True except ImportError: have_matplotlib = False class BaseProbplotMixin(object): def base_setup(self): if have_matplotlib: self.fig, self.ax = plt.subplots() self.other_array = np.random.normal(size=self.prbplt.data.shape) self.other_prbplot = sm.ProbPlot(self.other_array) def teardown(self): if have_matplotlib: plt.close('all') @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot(self): self.fig = self.prbplt.qqplot(ax=self.ax, line=self.line) @skipif(not have_matplotlib, reason='matplotlib not available') def test_ppplot(self): plt.close('all') self.fig = self.prbplt.ppplot(ax=self.ax, line=self.line) @skipif(not have_matplotlib, reason='matplotlib not available') def test_probplot(self): self.fig = self.prbplt.probplot(ax=self.ax, line=self.line) @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_other_array(self): self.fig = self.prbplt.qqplot(ax=self.ax, line=self.line, other=self.other_array) @skipif(not have_matplotlib, reason='matplotlib not available') def test_ppplot_other_array(self): self.fig = self.prbplt.ppplot(ax=self.ax, line=self.line, other=self.other_array) @skipif(not have_matplotlib, reason='matplotlib not available') def t_est_probplot_other_array(self): self.fig = self.prbplt.probplot(ax=self.ax, line=self.line, other=self.other_array) @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_other_prbplt(self): self.fig = self.prbplt.qqplot(ax=self.ax, line=self.line, other=self.other_prbplot) @skipif(not have_matplotlib, reason='matplotlib not available') def test_ppplot_other_prbplt(self): self.fig = self.prbplt.ppplot(ax=self.ax, line=self.line, other=self.other_prbplot) @skipif(not have_matplotlib, reason='matplotlib not available') def t_est_probplot_other_prbplt(self): self.fig = self.prbplt.probplot(ax=self.ax, line=self.line, other=self.other_prbplot) @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_custom_labels(self): self.fig = self.prbplt.qqplot(ax=self.ax, line=self.line, xlabel='Custom X-Label', ylabel='Custom Y-Label') @skipif(not have_matplotlib, reason='matplotlib not available') def test_ppplot_custom_labels(self): self.fig = self.prbplt.ppplot(ax=self.ax, line=self.line, xlabel='Custom X-Label', ylabel='Custom Y-Label') @skipif(not have_matplotlib, reason='matplotlib not available') def test_probplot_custom_labels(self): self.fig = self.prbplt.probplot(ax=self.ax, line=self.line, xlabel='Custom X-Label', ylabel='Custom Y-Label') @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_pltkwargs(self): self.fig = self.prbplt.qqplot(ax=self.ax, line=self.line, marker='d', markerfacecolor='cornflowerblue', markeredgecolor='white', alpha=0.5) @skipif(not have_matplotlib, reason='matplotlib not available') def test_ppplot_pltkwargs(self): self.fig = self.prbplt.ppplot(ax=self.ax, line=self.line, marker='d', markerfacecolor='cornflowerblue', markeredgecolor='white', alpha=0.5) @skipif(not have_matplotlib, reason='matplotlib not available') def test_probplot_pltkwargs(self): self.fig = self.prbplt.probplot(ax=self.ax, line=self.line, marker='d', markerfacecolor='cornflowerblue', markeredgecolor='white', alpha=0.5) class TestProbPlotLongely(BaseProbplotMixin): def setup(self): np.random.seed(5) self.data = sm.datasets.longley.load() self.data.exog = sm.add_constant(self.data.exog, prepend=False) self.mod_fit = sm.OLS(self.data.endog, self.data.exog).fit() self.prbplt = sm.ProbPlot(self.mod_fit.resid, stats.t, distargs=(4,)) self.line = 'r' self.base_setup() class TestProbPlotRandomNormalMinimal(BaseProbplotMixin): def setup(self): np.random.seed(5) self.data = np.random.normal(loc=8.25, scale=3.25, size=37) self.prbplt = sm.ProbPlot(self.data) self.line = None self.base_setup() class TestProbPlotRandomNormalWithFit(BaseProbplotMixin): def setup(self): np.random.seed(5) self.data = np.random.normal(loc=8.25, scale=3.25, size=37) self.prbplt = sm.ProbPlot(self.data, fit=True) self.line = 'q' self.base_setup() class TestProbPlotRandomNormalLocScale(BaseProbplotMixin): def setup(self): np.random.seed(5) self.data = np.random.normal(loc=8.25, scale=3.25, size=37) self.prbplt = sm.ProbPlot(self.data, loc=8.25, scale=3.25) self.line = '45' self.base_setup() class TestTopLevel(object): def setup(self): self.data = sm.datasets.longley.load() self.data.exog = sm.add_constant(self.data.exog, prepend=False) self.mod_fit = sm.OLS(self.data.endog, self.data.exog).fit() self.res = self.mod_fit.resid self.prbplt = sm.ProbPlot(self.mod_fit.resid, stats.t, distargs=(4,)) self.other_array = np.random.normal(size=self.prbplt.data.shape) self.other_prbplot = sm.ProbPlot(self.other_array) def teardown(self): if have_matplotlib: plt.close('all') @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot(self): fig = sm.qqplot(self.res, line='r') @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_pltkwargs(self): fig = sm.qqplot(self.res, line='r', marker='d', markerfacecolor='cornflowerblue', markeredgecolor='white', alpha=0.5) @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_2samples_ProbPlotObjects(self): # also tests all values for line for line in ['r', 'q', '45', 's']: # test with `ProbPlot` instances fig = sm.qqplot_2samples(self.prbplt, self.other_prbplot, line=line) plt.close('all') @skipif(not have_matplotlib, reason='matplotlib not available') def test_qqplot_2samples_arrays(self): # also tests all values for line for line in ['r', 'q', '45', 's']: # test with arrays fig = sm.qqplot_2samples(self.res, self.other_array, line=line) plt.close('all')