Modify tests for heatmap.js