Fix Flake8 validation issues.

Author: igorborgest

testing/test_awswrangler/test_data_lake.py

@@ -62,7 +62,7 @@ def external_schema(cloudformation_outputs, database):
     IAM_ROLE '{cloudformation_outputs["RedshiftRole"]}'
     REGION '{region}';
     """
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     with engine.connect() as con:
         con.execute(sql)
     yield "aws_data_wrangler_external"
@@ -254,7 +254,7 @@ def test_athena_ctas(bucket, database, kms_key):
     # keep_files=False
     wr.s3.delete_objects(path=s3_output)
     dfs = wr.athena.read_sql_query(
-        sql=f"SELECT * FROM test_athena_ctas",
+        sql="SELECT * FROM test_athena_ctas",
         database=database,
         ctas_approach=True,
         chunksize=1,
@@ -272,7 +272,7 @@ def test_athena_ctas(bucket, database, kms_key):
     # keep_files=True
     wr.s3.delete_objects(path=s3_output)
     dfs = wr.athena.read_sql_query(
-        sql=f"SELECT * FROM test_athena_ctas",
+        sql="SELECT * FROM test_athena_ctas",
         database=database,
         ctas_approach=True,
         chunksize=2,
@@ -720,7 +720,7 @@ def test_athena_query_failed(database):
 
 def test_athena_read_list(database):
     with pytest.raises(wr.exceptions.UnsupportedType):
-        wr.athena.read_sql_query(sql=f"SELECT ARRAY[1, 2, 3]", database=database, ctas_approach=False)
+        wr.athena.read_sql_query(sql="SELECT ARRAY[1, 2, 3]", database=database, ctas_approach=False)
 
 
 def test_sanitize_names():

testing/test_awswrangler/test_db.py

@@ -71,7 +71,7 @@ def external_schema(cloudformation_outputs, parameters, glue_database):
     IAM_ROLE '{parameters["redshift"]["role"]}'
     REGION '{region}';
     """
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     with engine.connect() as con:
         con.execute(sql)
     yield "aws_data_wrangler_external"
@@ -160,7 +160,7 @@ def test_redshift_temp_engine(parameters):
 
 
 def test_postgresql_param():
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-postgresql")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-postgresql")
     df = wr.db.read_sql_query(sql="SELECT %(value)s as col0", con=engine, params={"value": 1})
     assert df["col0"].iloc[0] == 1
     df = wr.db.read_sql_query(sql="SELECT %s as col0", con=engine, params=[1])
@@ -170,7 +170,7 @@ def test_postgresql_param():
 def test_redshift_copy_unload(bucket, parameters):
     path = f"s3://{bucket}/test_redshift_copy/"
     df = get_df().drop(["iint8", "binary"], axis=1, inplace=False)
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     wr.db.copy_to_redshift(
         df=df,
         path=path,
@@ -220,7 +220,7 @@ def test_redshift_copy_unload(bucket, parameters):
 
 
 def test_redshift_copy_upsert(bucket, parameters):
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     df = pd.DataFrame({"id": list((range(1_000))), "val": list(["foo" if i % 2 == 0 else "boo" for i in range(1_000)])})
     df3 = pd.DataFrame(
         {"id": list((range(1_000, 1_500))), "val": list(["foo" if i % 2 == 0 else "boo" for i in range(500)])}
@@ -313,7 +313,7 @@ def test_redshift_copy_upsert(bucket, parameters):
 )
 def test_redshift_exceptions(bucket, parameters, diststyle, distkey, sortstyle, sortkey, exc):
     df = pd.DataFrame({"id": [1], "name": "joe"})
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     path = f"s3://{bucket}/test_redshift_exceptions_{random.randint(0, 1_000_000)}/"
     with pytest.raises(exc):
         wr.db.copy_to_redshift(
@@ -347,7 +347,7 @@ def test_redshift_spectrum(bucket, glue_database, external_schema):
         partition_cols=["par_int"],
     )["paths"]
     wr.s3.wait_objects_exist(paths=paths, use_threads=False)
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     with engine.connect() as con:
         cursor = con.execute(f"SELECT * FROM {external_schema}.test_redshift_spectrum")
         rows = cursor.fetchall()
@@ -361,7 +361,7 @@ def test_redshift_spectrum(bucket, glue_database, external_schema):
 def test_redshift_category(bucket, parameters):
     path = f"s3://{bucket}/test_redshift_category/"
     df = get_df_category().drop(["binary"], axis=1, inplace=False)
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     wr.db.copy_to_redshift(
         df=df,
         path=path,
@@ -399,7 +399,7 @@ def test_redshift_unload_extras(bucket, parameters, kms_key_id):
     schema = parameters["redshift"]["schema"]
     path = f"s3://{bucket}/{table}/"
     wr.s3.delete_objects(path=path)
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-redshift")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-redshift")
     df = pd.DataFrame({"id": [1, 2], "name": ["foo", "boo"]})
     wr.db.to_sql(df=df, con=engine, name=table, schema=schema, if_exists="replace", index=False)
     paths = wr.db.unload_redshift_to_files(
@@ -466,7 +466,7 @@ def test_to_sql_cast(parameters, db_type):
 def test_uuid(parameters):
     table = "test_uuid"
     schema = parameters["postgresql"]["schema"]
-    engine = wr.catalog.get_engine(connection=f"aws-data-wrangler-postgresql")
+    engine = wr.catalog.get_engine(connection="aws-data-wrangler-postgresql")
     df = pd.DataFrame(
         {
             "id": [1, 2, 3],