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)