import unittest

import plotly_express as px

class TestPlotlyExpress(unittest.TestCase):

def test_basic_scatter_plot(self):

gapminder = px.data.gapminder()

gapminder2007 = gapminder.query("year == 2007")

px.scatter(gapminder2007, x="gdpPercap", y="lifeExp")

def test_complex_scatter_plot(self):

gapminder = px.data.gapminder()

gapminder2007 = gapminder.query("year == 2007")

px.scatter(gapminder, x="gdpPercap", y="lifeExp",size="pop", size_max=60, color="continent", hover_name="country",

animation_frame="year", animation_group="country", log_x=True, range_x=[100,100000], range_y=[25,90],

labels=dict(pop="Population", gdpPercap="GDP per Capita", lifeExp="Life Expectancy"))

def test_choropleth_plot(self):

gapminder = px.data.gapminder()

gapminder2007 = gapminder.query("year == 2007")

px.choropleth(gapminder, locations="iso_alpha", color="lifeExp", hover_name="country", animation_frame="year",

color_continuous_scale=px.colors.sequential.Plasma, projection="natural earth")

def test_violin_plot(self):

tips = px.data.tips()

px.scatter(tips, x="total_bill", y="tip", color="smoker", trendline="ols", marginal_x="violin", marginal_y="box")