def test_not_main(self, mock_resolve):
mock_resolve.return_value = self.data
mfp.add_cum_kappa(self.ax,
self.data,
direction='x',
main=False,
scale=False,
fill=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
mfp.add_cum_kappa(self.ax,
self.data,
direction='x',
main=True,
scale=False,
fill=True)
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()
self.ax.set_ylim.assert_called_once()
mock_resolve.assert_called_once()
def setUpClass(cls):
cls.d = {'mean_free_path': [[[[[0, 1, 2, 3]]]]],
'mode_kappa': [[[[[0, 2, 11, 100]]]]],
'temperature': [11],
'meta': {}}
cls.fig, cls.ax = one_small_legend()
cls.ax = mfp.add_cum_kappa(cls.ax, cls.d, direction='x', kmin=10,
main=True, scale=False)
def test_markers(self, mock_resolve):
mock_resolve.return_value = self.data
mfp.add_cum_kappa(self.ax,
self.data,
direction='x',
main=True,
scale=False,
fill=False,
xmarkers=1,
ymarkers=1)
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_called_once()
self.ax.set_ylim.assert_called_once()
mock_resolve.assert_called_once()
def setUpClass(cls):
cls.d = {'mean_free_path': [[[[[0, 2, 4]]]]],
'mode_kappa': [[[[[0, 1, 2]]]]],
'temperature': [11],
'meta': {}}
cls.fig, cls.ax = one_small_legend()
cls.ylim = (1, 2)
cls.ax.set_ylim(cls.ylim)
cls.ax = mfp.add_cum_kappa(cls.ax, cls.d, direction='x',
main=False, scale=True)