def test_from_sql(make_sql_connection): filename = "test_from_sql.db" table = "test_from_sql" conn = make_sql_connection(filename, table) query = "select * from {0}".format(table) pandas_df = pandas.read_sql(query, conn) modin_df = pd.read_sql(query, conn) assert modin_df_equals_pandas(modin_df, pandas_df) pandas_df = pandas.read_sql(query, conn, index_col="index") modin_df = pd.read_sql(query, conn, index_col="index") assert modin_df_equals_pandas(modin_df, pandas_df) with pytest.warns(UserWarning): pd.read_sql_query(query, conn) with pytest.warns(UserWarning): pd.read_sql_table(table, conn) # Test SQLAlchemy engine conn = sa.create_engine(conn) pandas_df = pandas.read_sql(query, conn) modin_df = pd.read_sql(query, conn) assert modin_df_equals_pandas(modin_df, pandas_df) # Test SQLAlchemy Connection conn = conn.connect() pandas_df = pandas.read_sql(query, conn) modin_df = pd.read_sql(query, conn) assert modin_df_equals_pandas(modin_df, pandas_df)
def test_from_sql(make_sql_connection): filename = "test_from_sql.db" table = "test_from_sql" conn = make_sql_connection(filename, table) query = "select * from {0}".format(table) pandas_df = pandas.read_sql(query, conn) modin_df = pd.read_sql(query, conn) assert modin_df_equals_pandas(modin_df, pandas_df) with pytest.warns(UserWarning): pd.read_sql_query(query, conn) with pytest.warns(UserWarning): pd.read_sql_table(table, conn)