@viz_reg_test
def test_heatmap():
# Compute x^2 + y^2 across a 2D grid
x, y = np.meshgrid(range(-5, 6), range(-5, 6))
z = x**2 + y**2
# Convert this grid to columnar data expected by Altair
data = pd.DataFrame({
"x": x.ravel(),
"y": y.ravel(),
"z": z.ravel(),
})
return ar.heatmap(data, x="x", y="y", color="z")
show_test(test_heatmap)
@viz_reg_test
def test_count_heatmap():
source = data.movies.url
return ar.count_heatmap(
source,
x="IMDB_Rating",
y="Rotten_Tomatoes_Rating",
)
show_test(test_count_heatmap)