def test_abbrev_title():
long_content = "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum."
dataset = [
{
"long_attr": long_content,
"normal": 3,
"normal2": 1
},
{
"long_attr": long_content,
"normal": 3,
"normal2": 1
},
{
"long_attr": long_content,
"normal": 2,
"normal2": 1
},
{
"long_attr": long_content,
"normal": 4,
"normal2": 1
},
]
df = pd.DataFrame(dataset)
lux.config.plotting_backend = "matplotlib"
vis = Vis(["normal2", "normal", f"long_attr={long_content}"], df)
vis_code = vis.to_matplotlib()
print(vis_code)
assert "long_attr = Lor...t laborum.'" in vis_code
vis_code = vis.to_altair()
print(vis_code)
assert "long_attr = Lor...t laborum.'" in vis_code
lux.config.plotting_backend = "altair"
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()
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, step=38.7), 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()
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()
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()
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_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()
assert "ax.set_xlabel('Acceleration')" in vis_code
assert "ax.set_ylabel('Origin')" 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()
assert "ax.scatter(x_pts, y_pts, alpha=0.5)" in vis_code
assert ("ax.set_xlabel('Acceleration', fontsize='15')" in vis_code
or "ax.set_xlabel('Acceleration')" in vis_code)
assert "ax.set_ylabel('Weight', fontsize='15')" in vis_code or "ax.set_ylabel('Weight')" in vis_code
