def test_plot_wrong_label_type(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with pytest.raises(TypeError):
with warnings.catch_warnings(record=True) as w:
cr.plot(labels=123)
assert np.any([
"must be either a list or tuple" in str(wi.message)
for wi in w
])
def test_plot_custom_filename(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(save=True, filename='cr.png')
assert os.path.isfile('cr.png')
os.unlink('cr.png')
def test_plot_default_filename(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(save=True, title="Correlation")
assert os.path.isfile('corr.pdf')
os.unlink('corr.pdf')
def test_plot_title(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(title="Test for Cross Correlation")
assert plt.fignum_exists(1)
def test_plot_axis(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(axis=[0, 1, 0, 100])
assert plt.fignum_exists(1)
def test_plot_labels_index_error(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with warnings.catch_warnings(record=True) as w:
cr.plot(labels='x')
assert np.any(
["must have two labels" in str(wi.message) for wi in w])
def test_simple_plot(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot()
assert plt.fignum_exists(1)
def test_plot_custom_filename(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(save=True, filename='cr.png')
assert os.path.isfile('cr.png')
os.unlink('cr.png')
def test_plot_matplotlib_not_installed(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with pytest.raises(ImportError) as excinfo:
cr.plot()
message = str(excinfo.value)
assert "Matplotlib required for plot()" in message
def test_plot_title(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(title="Test for Cross Correlation")
assert plt.fignum_exists(1)
def test_plot_default_filename(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(save=True)
assert os.path.isfile('corr.png')
os.unlink('corr.png')
def test_plot_axis(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot(axis=[0, 1, 0, 100])
assert plt.fignum_exists(1)
def test_plot_labels_index_error(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with warnings.catch_warnings(record=True) as w:
cr.plot(labels='x')
assert "must have two labels" in str(w[0].message)
def test_plot_wrong_label_type(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with pytest.raises(TypeError):
with warnings.catch_warnings(record=True) as w:
cr.plot(labels=123)
assert "must be either a list or tuple" in str(w[0].message)
def test_simple_plot(self):
cr = CrossCorrelation(self.lc1, self.lc2)
cr.plot()
assert plt.fignum_exists(1)
def test_plot_matplotlib_not_installed(self):
cr = CrossCorrelation(self.lc1, self.lc2)
with pytest.raises(ImportError) as excinfo:
cr.plot()
message = str(excinfo.value)
assert "Matplotlib required for plot()" in message
