def test_histogram_uniform():
df = pd.read_csv("lux/data/car.csv")
df["Year"] = pd.to_datetime(df["Year"], format="%Y")
df["Units"] = 4.0
vis = Vis(["Units"], df)
vis_code = vis.to_altair()
assert "y = alt.Y('Units', type= 'nominal'" in vis_code
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_vis_to_altair_custom_named_df(global_var):
df = pytest.car_df
some_weirdly_named_df = df.dropna()
vis = Vis(["Weight", "Horsepower"], some_weirdly_named_df)
code = vis.to_altair()
assert ("alt.Chart(some_weirdly_named_df)"
in code), "Unable to export to Altair and detect custom df name"
def test_bar_uniform():
df = pd.read_csv("lux/data/car.csv")
df["Year"] = pd.to_datetime(df["Year"], format="%Y")
df["Type"] = "A"
vis = Vis(["Type"], df)
vis_code = vis.to_altair()
assert "y = alt.Y('Type', type= 'nominal'" in vis_code
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_text_not_overridden():
from lux.vis.Vis import Vis
df = pd.read_csv("lux/data/college.csv")
vis = Vis(["Region", "Geography"], df)
vis._ipython_display_()
code = vis.to_altair()
assert 'color = "#ff8e04"' in code
def test_vis_to_altair_standalone(global_var):
df = pytest.car_df
vis = Vis(["Weight", "Horsepower"], df)
code = vis.to_altair(standalone=True)
assert (
"chart = alt.Chart(pd.DataFrame({'Weight': {0: 3504, 1: 3693, 2: 3436, 3: 3433, 4: 3449, 5: 43"
in code or
"alt.Chart(pd.DataFrame({'Horsepower': {0: 130, 1: 165, 2: 150, 3: 150, 4: 140,"
in code)
def test_period_to_altair(global_var):
df = pd.read_csv("lux/data/car.csv")
df["Year"] = pd.to_datetime(df["Year"], format="%Y")
df["Year"] = pd.DatetimeIndex(df["Year"]).to_period(freq="A")
from lux.vis.Vis import Vis
vis = Vis(["Acceleration", "Horsepower", "Year=1972"], df)
exported_code = vis.to_altair()
assert "Year = 1972" in exported_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
def test_vis_to_altair_basic_df(global_var):
df = pytest.car_df
vis = Vis(["Weight", "Horsepower"], df)
code = vis.to_altair()
assert "alt.Chart(df)" in code, "Unable to export to Altair"
def test_special_char():
dataset = [
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 5
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 3
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 6
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 7
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 3,
"normal": 10
},
{
"special.char": 1,
"normal": 1
},
{
"special.char": 5,
"normal": 2
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 2
},
{
"special.char": 1,
"normal": 2
},
]
test = pd.DataFrame(dataset)
from lux.vis.Vis import Vis
# TODO: add assert that checks that the bar chart is rendered correctly in Altair
vis = Vis(["special.char"], test)
assert vis.mark == "bar"
assert vis.intent == ["special.char"]
assert vis.get_attr_by_channel("x")[0].attribute == "Record"
assert vis.get_attr_by_channel("y")[0].attribute == "special.char"
vis = vis.to_altair()
assert (
"alt.Y('specialchar', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='special.char'))"
in vis)
assert (
"alt.X('Record', type= 'quantitative', title='Number of Records', axis=alt.Axis(title='Number of Records')"
in vis)
# Checking that this works even when there are multiple "." in column
test = test.rename(columns={"special.char": "special..char.."})
# TODO: add assert that checks that the bar chart is rendered correctly in Altair
vis = Vis(["special..char.."], test)
assert vis.mark == "bar"
assert vis.intent == ["special..char.."]
assert vis.get_attr_by_channel("x")[0].attribute == "Record"
assert vis.get_attr_by_channel("y")[0].attribute == "special..char.."
vis = vis.to_altair()
assert (
"alt.Y('specialchar', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='special..char..')"
in vis)
assert (
"alt.X('Record', type= 'quantitative', title='Number of Records', axis=alt.Axis(title='Number of Records')"
in vis)
