def test_from_sql_defaults(make_sql_connection): # noqa: F811
filename = "test_from_sql_distributed.db"
table = "test_from_sql_distributed"
conn = make_sql_connection(filename, table)
query = "select * from {0}".format(table)
pandas_df = pandas.read_sql(query, conn)
with pytest.warns(UserWarning):
modin_df_from_query = pd.read_sql(query, conn)
with pytest.warns(UserWarning):
modin_df_from_table = pd.read_sql(table, conn)
assert modin_df_equals_pandas(modin_df_from_query, pandas_df)
assert modin_df_equals_pandas(modin_df_from_table, pandas_df)
def test_from_sql_distributed(make_sql_connection): # noqa: F811
if Engine.get() == "Ray":
pytest.xfail("Distributed read_sql is broken, see GH#2194")
filename = "test_from_sql_distributed.db"
table = "test_from_sql_distributed"
conn = make_sql_connection(filename, table)
query = "select * from {0}".format(table)
pandas_df = pandas.read_sql(query, conn)
modin_df_from_query = pd.read_sql(
query, conn, partition_column="col1", lower_bound=0, upper_bound=6
)
modin_df_from_table = pd.read_sql(
table, conn, partition_column="col1", lower_bound=0, upper_bound=6
)
df_equals(modin_df_from_query, pandas_df)
df_equals(modin_df_from_table, pandas_df)
def test_from_sql_distributed(make_sql_connection): # noqa: F811
if os.environ.get("MODIN_ENGINE", "") == "Ray":
filename = "test_from_sql_distributed.db"
table = "test_from_sql_distributed"
conn = make_sql_connection(filename, table)
query = "select * from {0}".format(table)
pandas_df = pandas.read_sql(query, conn)
modin_df_from_query = pd.read_sql(query,
conn,
partition_column="col1",
lower_bound=0,
upper_bound=6)
modin_df_from_table = pd.read_sql(table,
conn,
partition_column="col1",
lower_bound=0,
upper_bound=6)
assert modin_df_equals_pandas(modin_df_from_query, pandas_df)
assert modin_df_equals_pandas(modin_df_from_table, pandas_df)
