def test_populate_options(global_var): from lux.processor.Compiler import Compiler df = pytest.car_df df.set_intent( [lux.Clause(attribute="?"), lux.Clause(attribute="MilesPerGal")]) col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal(list(col_set), list(df.columns)) df.set_intent([ lux.Clause(attribute="?", data_model="measure"), lux.Clause(attribute="MilesPerGal"), ]) df._ipython_display_() col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal( list(col_set), [ "Acceleration", "Weight", "Horsepower", "MilesPerGal", "Displacement" ], ) df.clear_intent()
def test_populate_options(): from lux.processor.Compiler import Compiler df = pd.read_csv("lux/data/car.csv") df.set_intent( [lux.Clause(attribute="?"), lux.Clause(attribute="MilesPerGal")]) col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal(list(col_set), list(df.columns)) df.set_intent([ lux.Clause(attribute="?", data_model="measure"), lux.Clause(attribute="MilesPerGal"), ]) df._repr_html_() col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal( list(col_set), [ "Acceleration", "Weight", "Horsepower", "MilesPerGal", "Displacement" ], )
def test_populate_options(global_var): from lux.processor.Compiler import Compiler # test for sql executor sql_df = lux.LuxSQLTable(table_name="cars") sql_df.set_intent([lux.Clause(attribute="?"), lux.Clause(attribute="milespergal")]) col_set = set() for specOptions in Compiler.populate_wildcard_options(sql_df._intent, sql_df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal(list(col_set), list(sql_df.columns)) sql_df.set_intent( [ lux.Clause(attribute="?", data_model="measure"), lux.Clause(attribute="milespergal"), ] ) sql_df._repr_html_() col_set = set() for specOptions in Compiler.populate_wildcard_options(sql_df._intent, sql_df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal( list(col_set), ["acceleration", "weight", "horsepower", "milespergal", "displacement"], )
def test_populate_options(global_var): lux.config.set_executor_type("Pandas") from lux.processor.Compiler import Compiler df = pytest.car_df df.set_intent( [lux.Clause(attribute="?"), lux.Clause(attribute="MilesPerGal")]) col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal(list(col_set), list(df.columns)) df.set_intent([ lux.Clause(attribute="?", data_model="measure"), lux.Clause(attribute="MilesPerGal"), ]) df._repr_html_() col_set = set() for specOptions in Compiler.populate_wildcard_options(df._intent, df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal( list(col_set), [ "Acceleration", "Weight", "Horsepower", "MilesPerGal", "Displacement" ], ) df.clear_intent() # test for sql executor connection = psycopg2.connect( "host=localhost dbname=postgres user=postgres password=lux") lux.config.set_SQL_connection(connection) sql_df = lux.LuxSQLTable(table_name="cars") sql_df.set_intent( [lux.Clause(attribute="?"), lux.Clause(attribute="milespergal")]) col_set = set() for specOptions in Compiler.populate_wildcard_options( sql_df._intent, sql_df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal(list(col_set), list(sql_df.columns)) sql_df.set_intent([ lux.Clause(attribute="?", data_model="measure"), lux.Clause(attribute="milespergal"), ]) sql_df._repr_html_() col_set = set() for specOptions in Compiler.populate_wildcard_options( sql_df._intent, sql_df)["attributes"]: for clause in specOptions: col_set.add(clause.attribute) assert list_equal( list(col_set), [ "acceleration", "weight", "horsepower", "milespergal", "displacement" ], )