# Get Python six functionality:
from __future__ import\
    absolute_import, print_function, division, unicode_literals


###############################################################################
###############################################################################
###############################################################################


import numpy as np
import pytest

import innvestigate.utils.visualizations as ivis


###############################################################################
###############################################################################
###############################################################################


@pytest.mark.fast
@pytest.mark.precommit
def test_fast__visualizations():
    def get_X():
        return np.random.rand(1, 28, 28, 3)

    ivis.project(get_X())
    ivis.heatmap(get_X())
    ivis.graymap(get_X())
    ivis.gamma(get_X())
    ivis.clip_quantile(get_X(), 0.95)