def test_plot_matplotlib_not_installed(self):
bs = Bispectrum(self.lc)
with pytest.raises(ImportError) as excinfo:
bs.plot_cum3()
bs.plot_mag()
bs.plot_phase()
message = str(excinfo.value)
assert "Matplotlib required for plot()" in message
def test_plot_matplotlib_not_installed(self):
bs = Bispectrum(self.lc)
with pytest.raises(ImportError) as excinfo:
bs.plot_cum3()
bs.plot_mag()
bs.plot_phase()
message = str(excinfo.value)
assert "Matplotlib required for plot()" in message
def test_plot_cum3_axis(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(axis=[0, 1, 0, 100])
assert plt.fignum_exists(1)
def test_plot_cum3(self):
bs = Bispectrum(self.lc)
bs.plot_cum3()
assert plt.fignum_exists(1)
def test_plot_cum3_custom_filename(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(save=True, filename='cum3.png')
assert os.path.isfile('cum3.png')
os.unlink('cum3.png')
def test_plot_cum3_default_filename(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(save=True)
assert os.path.isfile('bispec_cum3.png')
os.unlink('bispec_cum3.png')
def test_plot_cum3_axis(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(axis=[0, 1, 0, 100])
assert plt.fignum_exists(1)
def test_plot_cum3(self):
bs = Bispectrum(self.lc)
bs.plot_cum3()
assert plt.fignum_exists(1)
def test_plot_cum3_custom_filename(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(save=True, filename='cum3.png')
assert os.path.isfile('cum3.png')
os.unlink('cum3.png')
def test_plot_cum3_default_filename(self):
bs = Bispectrum(self.lc)
bs.plot_cum3(save=True)
assert os.path.isfile('bispec_cum3.png')
os.unlink('bispec_cum3.png')
