Write test for plot_heatmap

[rouille]

• Write tests for the function
• Add plot_show parameter and return the matplotlib.axes.Axes object of the plot (see postreise.plot.plot_tornado) so user can further modify the plot.