Update test/test_heatmap.rb