from kedro.extras.datasets.plotly import PlotlyDataSet
import plotly.express as px
import pandas as pd

df_data = pd.DataFrame([[0, 1], [1, 0]], columns=('x1', 'x2'))

data_set = PlotlyDataSet(
    filepath='scatter_plot.json',
    plotly_args={
        'type': 'scatter',
        'fig': {'x': 'x1', 'y': 'x2'},
    }
)
data_set.save(df_data)
reloaded = data_set.load()
assert px.scatter(df_data, x='x1', y='x2') == reloaded