def setUpClass(cls):
cls.d = {
'temperature': [1, 2],
'doping': [10, 100],
'zt': [[1, 2], [4, 8]]
}
cls.fig, cls.ax = one_colourbar()
cls.ax, cls.cbar = heatmap.add_ztmap(cls.ax, cls.d)
def test_precalculated(self, mock_colourbar, mock_resolve, mock_zt):
mock_resolve.return_value = self.data2
cbar = heatmap.add_ztmap(self.ax, self.data2, xinterp=None,
yinterp=None)
self.ax.pcolormesh.assert_called_once()
self.ax.set_xscale.assert_called_once_with('linear')
self.ax.set_yscale.assert_called_once_with('log')
self.ax.set_xlim.assert_called_once_with(1, 3)
self.ax.set_ylim.assert_called_once_with(10, 1000)
mock_colourbar.assert_called_once()
cbar.ax.set_yscale.assert_called_once_with('linear')
mock_resolve.assert_not_called()
mock_zt.assert_not_called()
def setUpClass(cls):
c = [[[[2.]], [[4.]]], [[[8.]], [[16.]]]]
s = [[[[3e6]], [[6e6]]], [[[9e6]], [[12e6]]]]
etc = [[[[3.]], [[3.]]], [[[3.]], [[3.]]]]
ltc = [[6.], [6.]]
t = [2., 4.]
d = [10., 100.]
cls.d = {
'conductivity': c,
'seebeck': s,
'electronic_thermal_conductivity': etc,
'lattice_thermal_conductivity': ltc,
'temperature': t,
'doping': d,
'meta': {
'units': {}
}
}
cls.fig, cls.ax = one_colourbar()
cls.ax, cls.cbar = heatmap.add_ztmap(cls.ax, cls.d, direction='x')