def test_heatmap_archive(_grid_archive, _cvt_archive, _sliding_archive):
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_grid_archive)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_cvt_archive)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive)
def test_heatmap_long_square(_long_grid_archive, _long_cvt_archive,
_long_sliding_archive):
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_long_grid_archive, square=True)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_long_cvt_archive, square=True)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_long_sliding_archive, square=True)
def test_cvt_archive_heatmap_with_custom_axis(_grid_archive, _cvt_archive,
_sliding_archive):
_, ax = plt.subplots(figsize=(8, 6))
grid_archive_heatmap(_grid_archive, ax=ax)
_, ax = plt.subplots(figsize=(8, 6))
cvt_archive_heatmap(_cvt_archive, ax=ax)
_, ax = plt.subplots(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive, ax=ax)
def test_heatmap_coolwarm_cmap(_grid_archive, _cvt_archive, _sliding_archive):
cmap = matplotlib.cm.get_cmap("coolwarm")
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_grid_archive, cmap=cmap)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_cvt_archive, cmap=cmap)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive, cmap=cmap)
def test_heatmap_listed_cmap(_grid_archive, _cvt_archive, _sliding_archive):
cmap = [[1, 0, 0], [0, 1, 0], [0, 0, 1]] # Primary red, green, blue.
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_grid_archive, cmap=cmap)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_cvt_archive, cmap=cmap)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive, cmap=cmap)
def test_heatmap_long_transpose(_long_grid_archive, _long_cvt_archive,
_long_sliding_archive):
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_long_grid_archive, transpose_bcs=True)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_long_cvt_archive, transpose_bcs=True)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_long_sliding_archive,
transpose_bcs=True)
def test_heatmap_with_limits(_grid_archive, _cvt_archive, _sliding_archive):
# Negative sphere function should have range (-2, 0). These limits should
# give a more uniform-looking archive.
kwargs = {"vmin": -1.0, "vmax": -0.5}
plt.figure(figsize=(8, 6))
grid_archive_heatmap(_grid_archive, **kwargs)
plt.figure(figsize=(8, 6))
cvt_archive_heatmap(_cvt_archive, **kwargs)
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive, **kwargs)
def test_sliding_archive_with_boundaries(_sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(_sliding_archive, boundary_lw=0.5)
def test_heatmap_listed_cmap__sliding(sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(sliding_archive,
cmap=[[1, 0, 0], [0, 1, 0], [0, 0, 1]])
def test_heatmap_coolwarm_cmap__sliding(sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(sliding_archive, cmap="coolwarm")
def test_heatmap_with_limits__sliding(sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(sliding_archive, vmin=-1.0, vmax=-0.5)
def test_heatmap_long_transpose__sliding(long_sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(long_sliding_archive,
transpose_bcs=True)
def test_heatmap_long_square__sliding(long_sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(long_sliding_archive, square=True)
def test_heatmap_with_custom_axis__sliding(sliding_archive):
_, ax = plt.subplots(figsize=(8, 6))
sliding_boundaries_archive_heatmap(sliding_archive, ax=ax)
def test_heatmap_archive__sliding(sliding_archive):
plt.figure(figsize=(8, 6))
sliding_boundaries_archive_heatmap(sliding_archive)