def test_save_sprite():
"""This test covers _save_sprite as well as _bytesIO_to_base64
"""
# Generate a simulated volume with a square inside
data = np.random.RandomState(0).uniform(size=140).reshape(7, 5, 4)
mask = np.zeros((7, 5, 4), dtype=int)
mask[1:-1, 1:-1, 1:-1] = 1
# Save the sprite using BytesIO
sprite_io = BytesIO()
html_stat_map._save_sprite(data,
sprite_io,
vmin=0,
vmax=1,
mask=mask,
format='png')
# Load the sprite back in base64
sprite_base64 = html_stat_map._bytesIO_to_base64(sprite_io)
decoded_io = BytesIO()
decoded_io.write(base64.b64decode(sprite_base64))
decoded_io.seek(0)
img = plt.imread(decoded_io, format="png")
correct_img = np.ma.array(html_stat_map._data_to_sprite(data),
mask=html_stat_map._data_to_sprite(mask))
correct_img = plt.Normalize(0, 1)(correct_img)
cmapped = plt.get_cmap("Greys")(correct_img)
assert np.allclose(img, cmapped, atol=.1)
def test_save_cmap():
"""This test covers _save_cmap as well as _bytesIO_to_base64
"""
# Save the cmap using BytesIO
cmap_io = BytesIO()
html_stat_map._save_cm(cmap_io, 'cold_hot', format='raw', n_colors=2)
# Load the colormap back in base64
cmap_base64 = html_stat_map._bytesIO_to_base64(cmap_io)
# Check the colormap is correct
assert cmap_base64 == '//////////8=\n'
def test_save_cmap():
"""This test covers _save_cmap as well as _bytesIO_to_base64
"""
# Save the cmap using BytesIO
cmap_io = BytesIO()
html_stat_map._save_cm(cmap_io, 'cold_hot', format='raw', n_colors=2)
# Load the colormap back in base64
cmap_base64 = html_stat_map._bytesIO_to_base64(cmap_io)
# Check the colormap is correct
assert cmap_base64 == '//////////8=\n'
def test_save_cmap():
"""This test covers _save_cmap as well as _bytesIO_to_base64
"""
# Save the cmap using BytesIO
cmap_io = BytesIO()
html_stat_map._save_cm(cmap_io, 'cold_hot', format='png', n_colors=2)
# Load the colormap back in base64
cmap_base64 = html_stat_map._bytesIO_to_base64(cmap_io)
# Check the colormap is correct
assert cmap_base64.startswith('iVBORw0KG')
assert cmap_base64.endswith('ElFTkSuQmCC')
def test_save_cmap(cmap, n_colors):
"""This test covers _save_cmap as well as _bytesIO_to_base64
"""
# Save the cmap using BytesIO
cmap_io = BytesIO()
html_stat_map._save_cm(cmap_io, cmap, format='png', n_colors=n_colors)
# Load the colormap back in base64
cmap_base64 = html_stat_map._bytesIO_to_base64(cmap_io)
decoded_io = BytesIO()
decoded_io.write(base64.b64decode(cmap_base64))
decoded_io.seek(0)
img = plt.imread(decoded_io, format="png")
expected = plt.get_cmap(cmap)(np.linspace(0, 1, n_colors))
assert np.allclose(img, expected, atol=.1)
def test_save_sprite():
"""This test covers _save_sprite as well as _bytesIO_to_base64
"""
# Generate a simulated volume with a square inside
data = np.zeros([2, 1, 1])
data[0, 0, 0] = 1
mask = data > 0
# Save the sprite using BytesIO
sprite_io = BytesIO()
html_stat_map._save_sprite(data, sprite_io, vmin=0, vmax=1,
mask=mask, format='raw')
# Load the sprite back in base64
sprite_base64 = html_stat_map._bytesIO_to_base64(sprite_io)
# Check the sprite is correct
assert sprite_base64 == '////AP////8=\n'
def test_save_sprite():
"""This test covers _save_sprite as well as _bytesIO_to_base64
"""
# Generate a simulated volume with a square inside
data = np.zeros([2, 1, 1])
data[0, 0, 0] = 1
mask = data > 0
# Save the sprite using BytesIO
sprite_io = BytesIO()
html_stat_map._save_sprite(data,
sprite_io,
vmin=0,
vmax=1,
mask=mask,
format='raw')
# Load the sprite back in base64
sprite_base64 = html_stat_map._bytesIO_to_base64(sprite_io)
# Check the sprite is correct
assert sprite_base64 == '////AP////8=\n'
