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