diff --git a/pandas/tests/io/excel/test_style.py b/pandas/tests/io/excel/test_style.py index 8c70e5eed3c97..12f14589365ff 100644 --- a/pandas/tests/io/excel/test_style.py +++ b/pandas/tests/io/excel/test_style.py @@ -1,5 +1,4 @@ import contextlib -import time import uuid import numpy as np @@ -324,18 +323,10 @@ def test_styler_to_s3(s3_bucket_public, s3so): target_file = f"{uuid.uuid4()}.xlsx" df = DataFrame({"x": [1, 2, 3], "y": [2, 4, 6]}) styler = df.style.set_sticky(axis="index") - styler.to_excel(f"s3://{mock_bucket_name}/{target_file}", storage_options=s3so) - timeout = 5 - while True: - if target_file in (obj.key for obj in s3_bucket_public.objects.all()): - break - time.sleep(0.1) - timeout -= 0.1 - assert timeout > 0, "Timed out waiting for file to appear on moto" - result = read_excel( - f"s3://{mock_bucket_name}/{target_file}", index_col=0, storage_options=s3so - ) - tm.assert_frame_equal(result, df) + uri = f"s3://{mock_bucket_name}/{target_file}" + styler.to_excel(uri, storage_options=s3so) + result = read_excel(uri, index_col=0, storage_options=s3so) + tm.assert_frame_equal(result, df) @pytest.mark.parametrize("merge_cells", [True, False, "columns"]) diff --git a/pandas/tests/io/json/test_pandas.py b/pandas/tests/io/json/test_pandas.py index a83460b24e48d..d67e725233127 100644 --- a/pandas/tests/io/json/test_pandas.py +++ b/pandas/tests/io/json/test_pandas.py @@ -7,7 +7,6 @@ import json import os import sys -import time import uuid import numpy as np @@ -2016,14 +2015,10 @@ def test_to_s3(self, s3_bucket_public, s3so): mock_bucket_name = s3_bucket_public.name target_file = f"{uuid.uuid4()}.json" df = DataFrame({"x": [1, 2, 3], "y": [2, 4, 6]}) - df.to_json(f"s3://{mock_bucket_name}/{target_file}", storage_options=s3so) - timeout = 5 - while True: - if target_file in (obj.key for obj in s3_bucket_public.objects.all()): - break - time.sleep(0.1) - timeout -= 0.1 - assert timeout > 0, "Timed out waiting for file to appear on moto" + uri = f"s3://{mock_bucket_name}/{target_file}" + df.to_json(uri, storage_options=s3so) + result = read_json(uri, storage_options=s3so) + tm.assert_frame_equal(result, df) def test_json_pandas_nulls(self, nulls_fixture): # GH 31615 diff --git a/pandas/tests/io/test_compression.py b/pandas/tests/io/test_compression.py index af250ced01d00..97b64a29a7f2c 100644 --- a/pandas/tests/io/test_compression.py +++ b/pandas/tests/io/test_compression.py @@ -5,7 +5,6 @@ import sys import tarfile import textwrap -import time import zipfile import numpy as np @@ -193,7 +192,6 @@ def test_gzip_reproducibility_file_name(temp_file): # test for filename path = temp_file df.to_csv(path, compression=compression_options) - time.sleep(0.1) output = path.read_bytes() df.to_csv(path, compression=compression_options) assert output == path.read_bytes() @@ -216,7 +214,6 @@ def test_gzip_reproducibility_file_object(): buffer = io.BytesIO() df.to_csv(buffer, compression=compression_options, mode="wb") output = buffer.getvalue() - time.sleep(0.1) buffer = io.BytesIO() df.to_csv(buffer, compression=compression_options, mode="wb") assert output == buffer.getvalue()