fix: Mark flaky throttled tests for rerun (#3266)

* fix: asynico context

* fix: Mark throttled tests for rerun

* Revert "fix: asynico context"

This reverts commit bdca559.

Author: kukushking

tests/unit/test_athena_iceberg.py

@@ -65,6 +65,7 @@ def test_athena_to_iceberg(
     assert_pandas_equals(df, df_out)
 
 
+@pytest.mark.flaky(reruns=5, reruns_delay=30)
 @pytest.mark.parametrize("partition_cols", [None, ["name"]])
 def test_athena_to_iceberg_append(
     path: str,

tests/unit/test_data_quality.py

@@ -74,6 +74,7 @@ def test_ruleset_dqdl(df, path, glue_database, glue_table, glue_ruleset, glue_da
     assert df_results["Result"].eq("PASS").all()
 
 
+@pytest.mark.flaky(reruns=5, reruns_delay=30)
 @pytest.mark.parametrize("name", [False, True])
 def test_recommendation_ruleset(df, path, name, glue_database, glue_table, glue_ruleset, glue_data_quality_role):
     df_recommended_ruleset = wr.data_quality.create_recommendation_ruleset(

tests/unit/test_quicksight.py

@@ -13,6 +13,7 @@
 client = boto3.client("quicksight")
 
 
+@pytest.mark.flaky(reruns=5, reruns_delay=30)
 @pytest.mark.xfail(raises=client.exceptions.ConflictException)
 def test_quicksight(
     path: str, quicksight_datasource: str, quicksight_dataset: str, glue_database: str, glue_table: str