heatmap: added simple tests