def test_log_scale(self, mock_log10):
mock_log10.return_value = [0, 1]
self.ax.get_ylim.return_value = [1, 10]
self.ax.get_yaxis().get_scale.return_value = 'log'
self.ax.reset_mock()
frequency.add_dos(self.ax,
self.data,
colour=self.colour,
total=False,
main=False,
scale=True,
fill=False,
line=True,
invert=False)
self.assertEqual(self.ax.plot.call_count, 2)
self.ax.set_xlabel.assert_not_called()
self.ax.set_ylabel.assert_not_called()
self.ax.set_xlim.assert_not_called()
self.ax.set_ylim.assert_not_called()
self.ax.get_xlim.assert_called_once()
self.ax.get_ylim.assert_called_once()
self.ax.get_xaxis().get_scale.assert_called_once()
self.ax.get_yaxis().get_scale.assert_called_once()
mock_log10.assert_called_once()
def test_total(self):
frequency.add_dos(self.ax, self.data, colour=self.colour, total=True,
main=True, scale=False, fill=False, line=True,
invert=False)
self.assertEqual(self.ax.plot.call_count, 3)
self.ax.set_xlabel.assert_called_once()
self.ax.set_ylabel.assert_called_once()
self.ax.set_xlim.assert_not_called()
self.ax.set_ylim.assert_called_once()
def setUpClass(cls):
cls.d = {
'frequency': [0, 1],
'A': [0, 3],
'B': [4, 0],
'total': [4, 3]
}
cls.c = {'A': '#ff0000', 'B': '#00ff0080', 'total': '#000000'}
cls.fig, cls.ax = one_small_legend()
cls.ax = frequency.add_dos(cls.ax,
cls.d,
cls.c,
main=True,
scale=False)
def setUpClass(cls):
cls.d = {
'frequency': [0, 1],
'A': [0, 3],
'B': [4, 0],
'total': [4, 3]
}
cls.c = {'A': '#ff0000', 'B': '#00ff0080', 'total': '#000000'}
cls.fig, cls.ax = one_small_legend()
cls.ylim = cls.ax.get_ylim()
cls.ax.set_yscale('log')
cls.ylim = [10, 100]
cls.ax.set_ylim(cls.ylim)
cls.ax = frequency.add_dos(cls.ax,
cls.d,
cls.c,
main=False,
scale=True)