def test_colored_heatmap_chart(global_var):
df = pd.read_csv(
"https://raw.githubusercontent.com/lux-org/lux-datasets/master/data/airbnb_nyc.csv"
)
lux.config.plotting_backend = "vegalite"
vis = Vis(["price", "longitude", "availability_365"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_rect()" in vis_code
assert (
"x=alt.X('xBinStart', type='quantitative', axis=alt.Axis(title='price'), bin = alt.BinParams(binned=True))"
in vis_code)
assert "x2=alt.X2('xBinEnd')" in vis_code
assert (
"y=alt.Y('yBinStart', type='quantitative', axis=alt.Axis(title='longitude'), bin = alt.BinParams(binned=True))"
in vis_code)
assert "y2=alt.Y2('yBinEnd')" in vis_code
assert 'scale=alt.Scale(type="log")' in vis_code
assert "chart.encode(color=alt.Color('availability_365',type='quantitative'))" in vis_code
lux.config.plotting_backend = "matplotlib"
vis = Vis(["price", "longitude", "availability_365"], df)
vis_code = vis.to_matplotlib_code()
assert "plt.imshow(df, cmap='viridis')" in vis_code
assert "index='xBinStart'" in vis_code
assert "values='availability_365'" in vis_code
assert "columns='yBinStart'" in vis_code
assert "plt.colorbar(label='availability_365')" in vis_code
def test_histogram_chart(global_var):
df = pytest.car_df
lux.config.plotting_backend = "vegalite"
vis = Vis(["Displacement"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_bar" in vis_code
assert (
"alt.X('Displacement', title='Displacement (binned)',bin=alt.Bin(binned=True), type='quantitative', axis=alt.Axis(labelOverlap=True, title='Displacement (binned)'), scale=alt.Scale(domain=(68.0, 455.0)))"
in vis_code)
assert 'alt.Y("Number of Records", type="quantitative")' in vis_code
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Displacement"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.bar(bars, measurements, width=32.25)" in vis_code
assert "ax.set_xlabel('Displacement (binned)')" in vis_code
assert "ax.set_ylabel('Number of Records')" in vis_code
def test_line_chart(global_var):
df = pytest.car_df
lux.config.plotting_backend = "vegalite"
vis = Vis(["Year", "Acceleration"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_line()" in vis_code
assert (
"y = alt.Y('Acceleration', type= 'quantitative', title='Mean of Acceleration', axis=alt.Axis(title='Acceleration')"
in vis_code)
assert "x = alt.X('Year', type = 'temporal', axis=alt.Axis(title='Year'))" in vis_code
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Year", "Acceleration"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.plot(x_pts, y_pts)" in vis_code
assert "ax.set_xlabel('Year')" in vis_code
assert "ax.set_ylabel('Mean of Acceleration')" in vis_code
def test_colored_line_chart(global_var):
df = pd.read_csv("lux/data/car.csv")
lux.config.plotting_backend = "vegalite"
vis = Vis(["Year", "Acceleration", "Origin"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_line()" in vis_code
assert (
"y = alt.Y('Acceleration', type= 'quantitative', title='Mean of Acceleration', axis=alt.Axis(title='Acceleration')"
in vis_code)
assert "x = alt.X('Year', type = 'temporal', axis=alt.Axis(title='Year'))" in vis_code
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Year", "Acceleration", "Origin"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.plot" in vis_code
assert "title='Origin'" in vis_code
assert "ax.set_xlabel('Year')" in vis_code
assert "ax.set_ylabel('Mean of Acceleration')" in vis_code
def test_scatter_chart(global_var):
df = pytest.car_df
lux.config.plotting_backend = "vegalite"
vis = Vis(["Acceleration", "Weight"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(df).mark_circle()" in vis_code
assert (
"x=alt.X('Acceleration',scale=alt.Scale(domain=(8.0, 24.8)),type='quantitative', axis=alt.Axis(title='Acceleration'))"
in vis_code)
assert (
" y=alt.Y('Weight',scale=alt.Scale(domain=(1613, 5140)),type='quantitative', axis=alt.Axis(title='Weight'))"
in vis_code)
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Acceleration", "Weight"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.scatter(x_pts, y_pts, alpha=0.5)" in vis_code
assert "ax.set_xlabel('Acceleration')" in vis_code
assert "ax.set_ylabel('Weight')" in vis_code
def test_colored_bar_chart(global_var):
df = pytest.car_df
lux.config.plotting_backend = "vegalite"
vis = Vis(["Cylinders", "Acceleration", "Origin"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_bar()" in vis_code
assert (
"y = alt.Y('Cylinders', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='Cylinders'))"
in vis_code)
assert (
"x = alt.X('Acceleration', type= 'quantitative', title='Mean of Acceleration', axis=alt.Axis(title='Mean of Acceleration')"
in vis_code)
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Cylinders", "Acceleration", "Origin"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.barh" in vis_code
assert "title='Origin'" in vis_code
assert "ax.set_xlabel('Acceleration')" in vis_code
assert "ax.set_ylabel('Cylinders')" in vis_code
def test_bar_chart(global_var):
df = pytest.car_df
lux.config.plotting_backend = "vegalite"
vis = Vis(["Origin", "Acceleration"], df)
vis_code = vis.to_Altair()
assert "alt.Chart(visData).mark_bar()" in vis_code
assert (
"y = alt.Y('Origin', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='Origin'))"
in vis_code)
assert (
"x = alt.X('Acceleration', type= 'quantitative', title='Mean of Acceleration', axis=alt.Axis(title='Mean of Acceleration'))"
in vis_code)
lux.config.plotting_style = None
lux.config.plotting_backend = "matplotlib"
vis = Vis(["Origin", "Acceleration"], df)
vis_code = vis.to_matplotlib_code()
assert "ax.barh(bars, measurements, align='center')" in vis_code
assert "ax.set_xlabel('Acceleration')" in vis_code
assert "ax.set_ylabel('Origin')" in vis_code
