def test_format_as_image_invalid_expl(mock_expl_imgarr, mock_expl_imgmode): # invalid type with pytest.raises(TypeError): format_as_image(mock_expl_imgarr) # invalid image mode with pytest.raises(ValueError): format_as_image(mock_expl_imgmode)
def test_format_as_image_notransparency(catdog_rgba, mock_expl): # heatmap with full transparency overlay = format_as_image(mock_expl, alpha_limit=0.0) assert_pixel_by_pixel_equal(overlay, catdog_rgba)
def test_format_as_image_noheatmap(catdog_rgba, mock_expl_noheatmap): # no heatmap overlay = format_as_image(mock_expl_noheatmap) assert_pixel_by_pixel_equal(overlay, catdog_rgba)
def test_format_as_image_invalid_expl(mock_expl_imgarr): with pytest.raises(TypeError): format_as_image(mock_expl_imgarr)