def test_log_scale(self, mock_resolve, mock_log10): mock_resolve.return_value = self.data 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_cum_kappa(self.ax, self.data, direction='x', main=False, scale=True, fill=False, invert=False) self.ax.plot.assert_called_once() 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_not_main(self, mock_resolve): mock_resolve.return_value = self.data frequency.add_cum_kappa(self.ax, self.data, direction='x', main=False, scale=False, fill=False, invert=False) self.ax.plot.assert_called_once() 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() mock_resolve.assert_called_once()
def test_fill(self, mock_resolve): mock_resolve.return_value = self.data frequency.add_cum_kappa(self.ax, self.data, direction='x', main=True, scale=False, fill=True, invert=False) self.ax.fill_between.assert_called_once() self.ax.set_xlabel.assert_called_once() self.ax.set_ylabel.assert_called_once() self.ax.set_xlim.assert_called_once_with(0, 1) self.ax.set_ylim.assert_called_once_with(0, 2) mock_resolve.assert_called_once()
def test_invert(self, mock_resolve): mock_resolve.return_value = self.data frequency.add_cum_kappa(self.ax, self.data, direction='x', main=True, scale=False, fill=False, invert=True) self.ax.plot.assert_called_once() self.ax.set_xlabel.assert_called_once() self.ax.set_ylabel.assert_not_called() self.ax.tick_params.assert_called_once() self.ax.set_xlim.assert_called_once_with(0, 2) self.ax.set_ylim.assert_called_once_with(0, 1) mock_resolve.assert_called_once()
def setUpClass(cls): cls.d = { 'frequency': [0, 1], 'mode_kappa': [[[[[0, 2]]]]], 'temperature': [11], 'meta': {} } cls.fig, cls.ax = one_small_legend() cls.ax = frequency.add_cum_kappa(cls.ax, cls.d, direction='x', main=True, scale=False)
def setUpClass(cls): cls.d = { 'frequency': [0, 1], 'mode_kappa': [[[[[0, 2]]]]], 'temperature': [11], 'meta': {} } 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_cum_kappa(cls.ax, cls.d, direction='x', main=False, scale=True)