Skip to content

Commit 96075b9

Browse files
committed
Bumping version to 0.0.20
1 parent 0df629a commit 96075b9

File tree

3 files changed

+35
-2
lines changed

3 files changed

+35
-2
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
> Utility belt to handle data on AWS.
44
5-
[![Release](https://img.shields.io/badge/release-0.0.19-brightgreen.svg)](https://pypi.org/project/awswrangler/)
5+
[![Release](https://img.shields.io/badge/release-0.0.20-brightgreen.svg)](https://pypi.org/project/awswrangler/)
66
[![Downloads](https://img.shields.io/pypi/dm/awswrangler.svg)](https://pypi.org/project/awswrangler/)
77
[![Python Version](https://img.shields.io/badge/python-3.6%20%7C%203.7-brightgreen.svg)](https://pypi.org/project/awswrangler/)
88
[![Documentation Status](https://readthedocs.org/projects/aws-data-wrangler/badge/?version=latest)](https://aws-data-wrangler.readthedocs.io/en/latest/?badge=latest)

awswrangler/__version__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
__title__ = "awswrangler"
22
__description__ = "Utility belt to handle data on AWS."
3-
__version__ = "0.0.19"
3+
__version__ = "0.0.20"
44
__license__ = "Apache License 2.0"

testing/test_awswrangler/test_redshift.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,39 @@ def test_to_redshift_spark(session, bucket, redshift_parameters, sample_name, mo
247247
assert len(list(dataframe.columns)) == len(list(rows[0]))
248248

249249

250+
def test_to_redshift_spark_big(session, bucket, redshift_parameters):
251+
dataframe = session.spark_session.createDataFrame(
252+
pd.DataFrame({
253+
"A": list(range(100_000)),
254+
"B": list(range(100_000)),
255+
"C": list(range(100_000))
256+
}))
257+
con = Redshift.generate_connection(
258+
database="test",
259+
host=redshift_parameters.get("RedshiftAddress"),
260+
port=redshift_parameters.get("RedshiftPort"),
261+
user="test",
262+
password=redshift_parameters.get("RedshiftPassword"),
263+
)
264+
session.spark.to_redshift(
265+
dataframe=dataframe,
266+
path=f"s3://{bucket}/redshift-load/",
267+
connection=con,
268+
schema="public",
269+
table="test",
270+
iam_role=redshift_parameters.get("RedshiftRole"),
271+
mode="overwrite",
272+
min_num_partitions=10,
273+
)
274+
cursor = con.cursor()
275+
cursor.execute("SELECT * from public.test")
276+
rows = cursor.fetchall()
277+
cursor.close()
278+
con.close()
279+
assert dataframe.count() == len(rows)
280+
assert len(list(dataframe.columns)) == len(list(rows[0]))
281+
282+
250283
@pytest.mark.parametrize(
251284
"sample_name,mode,factor,diststyle,distkey,exc,sortstyle,sortkey",
252285
[

0 commit comments

Comments
 (0)