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"
],
)
