from unittest import SkipTest

import holoviews as hv
import numpy as np
import pandas as pd

from holoviews.element.comparison import ComparisonTestCase


class TestPandasTransforms(ComparisonTestCase):
    def setUp(self):
        import hvplot.pandas  # noqa

    def test_pandas_transform(self):
        demo_df = pd.DataFrame({'value': np.random.randn(50), 'probability': np.random.rand(50)})
        percent = hv.dim('probability') * 100
        scatter = demo_df.hvplot.scatter(
            x='value', y='probability', transforms=dict(probability=percent)
        )
        self.assertEqual((scatter.data['probability']).values, demo_df['probability'].values * 100)


class TestXArrayTransforms(ComparisonTestCase):
    def setUp(self):
        try:
            import xarray as xr  # noqa
        except ImportError:
            raise SkipTest('xarray not available')
        import hvplot.xarray  # noqa

    def test_xarray_transform(self):
        import xarray as xr

        data = np.arange(0, 60).reshape(6, 10)
        x = np.arange(10)
        y = np.arange(6)
        da = xr.DataArray(data, coords={'y': y, 'x': x}, dims=('y', 'x'), name='value')
        img = da.hvplot.image(transforms=dict(value=hv.dim('value') * 10))
        self.assertEqual(img.data.value.data, da.data * 10)