Skip to content

Commit 65f6d2b

Browse files
sfc-gh-stakedaankit-bhatnagar167
authored andcommitted
SNOW-116313: Remove usage of old query result format parameter in tests
1 parent 7c2653f commit 65f6d2b

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

DESCRIPTION.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ Release Notes
5252

5353
- Add asn1crypto requirement to mitigate incompatibility change.
5454

55-
- v1.9.0(August 26,2019) **REMOVED from pypi**
55+
- v1.9.0(August 26,2019) **REMOVED from pypi due to dependency compatibility issues**
5656

5757
- Implement converter for all arrow data types in python connector extension
5858
- Fix arrow error when returning empty result using python connecter

test/test_arrow_pandas.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,7 @@
2323
except ImportError:
2424
no_arrow_iterator_ext = True
2525

26-
SQL_ENABLE_ARROW = "alter session set query_result_format='ARROW_FORCE';"
27-
SQL_ENABLE_ARROW_V2 = "alter session set python_connector_query_result_format='ARROW_FORCE';"
26+
SQL_ENABLE_ARROW = "alter session set python_connector_query_result_format='ARROW';"
2827

2928
EPSILON = 1e-8
3029

@@ -230,7 +229,6 @@ def test_semi_struct(conn_cnx):
230229
# fetch dataframe with new arrow support
231230
cursor_table = cnx_table.cursor()
232231
cursor_table.execute(SQL_ENABLE_ARROW)
233-
cursor_table.execute(SQL_ENABLE_ARROW_V2)
234232
cursor_table.execute(sql_text)
235233
df_new = cursor_table.fetch_pandas_all()
236234
col_new = df_new.iloc[0]
@@ -432,7 +430,6 @@ def validate_pandas(conn_cnx, sql, cases, col_count, method='one', data_type='fl
432430
# fetch dataframe with new arrow support
433431
cursor_table = cnx_table.cursor()
434432
cursor_table.execute(SQL_ENABLE_ARROW)
435-
cursor_table.execute(SQL_ENABLE_ARROW_V2)
436433
cursor_table.execute(sql)
437434

438435
# build dataframe
@@ -527,7 +524,6 @@ def test_empty(conn_cnx):
527524
with conn_cnx() as cnx:
528525
cursor = cnx.cursor()
529526
cursor.execute(SQL_ENABLE_ARROW)
530-
cursor.execute(SQL_ENABLE_ARROW_V2)
531527
cursor.execute("select seq4() from table(generator(rowcount=>1)) limit 0")
532528
assert cursor.fetch_pandas_all() is None, 'the result is not none'
533529

@@ -557,7 +553,6 @@ def fetch_pandas(conn_cnx, sql, row_count, col_count, method='one'):
557553
# fetch dataframe by fetching row by row
558554
cursor_row = cnx_row.cursor()
559555
cursor_row.execute(SQL_ENABLE_ARROW)
560-
cursor_row.execute(SQL_ENABLE_ARROW_V2)
561556
cursor_row.execute(sql)
562557

563558
# build dataframe
@@ -583,7 +578,6 @@ def fetch_pandas(conn_cnx, sql, row_count, col_count, method='one'):
583578
# fetch dataframe with new arrow support
584579
cursor_table = cnx_table.cursor()
585580
cursor_table.execute(SQL_ENABLE_ARROW)
586-
cursor_table.execute(SQL_ENABLE_ARROW_V2)
587581
cursor_table.execute(sql)
588582

589583
# build dataframe

0 commit comments

Comments
 (0)