Run the Heatmap function during tests