Skip to content

Commit 197f06f

Browse files
committed
Add redshift spectrum tests for unsigned integers.
1 parent 7d190b3 commit 197f06f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

testing/test_awswrangler/test_data_lake.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1369,7 +1369,7 @@ def test_copy_replacing_filename(bucket):
13691369
wr.s3.delete_objects(path=path2)
13701370

13711371

1372-
def test_unsigned_parquet(bucket, database):
1372+
def test_unsigned_parquet(bucket, database, external_schema):
13731373
path = f"s3://{bucket}/test_unsigned_parquet/"
13741374
table = "test_unsigned_parquet"
13751375
wr.s3.delete_objects(path=path)
@@ -1391,6 +1391,11 @@ def test_unsigned_parquet(bucket, database):
13911391
assert df.c0.sum() == (2 ** 8) - 1
13921392
assert df.c1.sum() == (2 ** 16) - 1
13931393
assert df.c2.sum() == (2 ** 32) - 1
1394+
engine = wr.catalog.get_engine("aws-data-wrangler-redshift")
1395+
df = wr.db.read_sql_table(con=engine, table=table, schema=external_schema)
1396+
assert df.c0.sum() == (2 ** 8) - 1
1397+
assert df.c1.sum() == (2 ** 16) - 1
1398+
assert df.c2.sum() == (2 ** 32) - 1
13941399

13951400
df = pd.DataFrame({"c0": [0, 0, (2 ** 64) - 1]})
13961401
df["c0"] = df.c0.astype("uint64")

0 commit comments

Comments
 (0)