bug fix

SMTG-Bham · Oct 25, 2022 · fd7b994 · fd7b994
1 parent 559159a
commit fd7b994
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/src/_build/doctrees/environment.pickle b/docs/src/_build/doctrees/environment.pickle
diff --git a/tests/test_plot/test_heatmap.py b/tests/test_plot/test_heatmap.py
@@ -172,8 +172,8 @@ def test_default(self, mock_colourbar, mock_resolve, mock_interpolate, mock_zt):
         mock_zt.return_value = self.data2
         mock_interpolate.return_value = (self.data, self.data)
         mock_resolve.return_value = self.data
-        cbar = heatmap.add_ztmap(self.ax, self.data, xinterp=None,
-                                 yinterp=None)
+        cbar = heatmap.add_ztmap(self.ax, self.data, kdata=self.data,
+                                 xinterp=None, yinterp=None)
 
         self.ax.pcolormesh.assert_called_once()
         self.ax.set_xscale.assert_called_once_with('linear')