TST: Replace ensure_clean utility function with the temp_file pytest fixture in 5 test files (#62481)

Co-authored-by: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com>

Author: jsngn

pandas/tests/io/formats/style/test_style.py

@@ -1597,7 +1597,6 @@ def test_no_empty_apply(mi_styler):
 
 
 @pytest.mark.parametrize("format", ["html", "latex", "string"])
-def test_output_buffer(mi_styler, format):
+def test_output_buffer(mi_styler, format, temp_file):
     # gh 47053
-    with tm.ensure_clean(f"delete_me.{format}") as f:
-        getattr(mi_styler, f"to_{format}")(f)
+    getattr(mi_styler, f"to_{format}")(temp_file)

pandas/tests/io/formats/test_to_latex.py

@@ -8,7 +8,6 @@
     DataFrame,
     Series,
 )
-import pandas._testing as tm
 
 pytest.importorskip("jinja2")
 
@@ -30,27 +29,24 @@ def df_short():
 
 
 class TestToLatex:
-    def test_to_latex_to_file(self, float_frame):
-        with tm.ensure_clean("test.tex") as path:
-            float_frame.to_latex(path)
-            with open(path, encoding="utf-8") as f:
-                assert float_frame.to_latex() == f.read()
+    def test_to_latex_to_file(self, float_frame, temp_file):
+        float_frame.to_latex(temp_file)
+        with open(temp_file, encoding="utf-8") as f:
+            assert float_frame.to_latex() == f.read()
 
-    def test_to_latex_to_file_utf8_with_encoding(self):
+    def test_to_latex_to_file_utf8_with_encoding(self, temp_file):
         # test with utf-8 and encoding option (GH 7061)
         df = DataFrame([["au\xdfgangen"]])
-        with tm.ensure_clean("test.tex") as path:
-            df.to_latex(path, encoding="utf-8")
-            with open(path, encoding="utf-8") as f:
-                assert df.to_latex() == f.read()
+        df.to_latex(temp_file, encoding="utf-8")
+        with open(temp_file, encoding="utf-8") as f:
+            assert df.to_latex() == f.read()
 
-    def test_to_latex_to_file_utf8_without_encoding(self):
+    def test_to_latex_to_file_utf8_without_encoding(self, temp_file):
         # test with utf-8 without encoding option
         df = DataFrame([["au\xdfgangen"]])
-        with tm.ensure_clean("test.tex") as path:
-            df.to_latex(path)
-            with open(path, encoding="utf-8") as f:
-                assert df.to_latex() == f.read()
+        df.to_latex(temp_file)
+        with open(temp_file, encoding="utf-8") as f:
+            assert df.to_latex() == f.read()
 
     def test_to_latex_tabular_with_index(self):
         df = DataFrame({"a": [1, 2], "b": ["b1", "b2"]})

pandas/tests/io/json/test_readlines.py

@@ -195,7 +195,7 @@ def test_readjson_each_chunk(request, lines_json_df, engine):
     assert chunks[1].shape == (1, 2)
 
 
-def test_readjson_chunks_from_file(request, engine):
+def test_readjson_chunks_from_file(request, engine, temp_file):
     if engine == "pyarrow":
         # GH 48893
         reason = (
@@ -204,41 +204,39 @@ def test_readjson_chunks_from_file(request, engine):
         )
         request.applymarker(pytest.mark.xfail(reason=reason, raises=ValueError))
 
-    with tm.ensure_clean("test.json") as path:
-        df = DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
-        df.to_json(path, lines=True, orient="records")
-        with read_json(path, lines=True, chunksize=1, engine=engine) as reader:
-            chunked = pd.concat(reader)
-        unchunked = read_json(path, lines=True, engine=engine)
-        tm.assert_frame_equal(unchunked, chunked)
+    df = DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
+    df.to_json(temp_file, lines=True, orient="records")
+    with read_json(temp_file, lines=True, chunksize=1, engine=engine) as reader:
+        chunked = pd.concat(reader)
+    unchunked = read_json(temp_file, lines=True, engine=engine)
+    tm.assert_frame_equal(unchunked, chunked)
 
 
 @pytest.mark.parametrize("chunksize", [None, 1])
-def test_readjson_chunks_closes(chunksize):
-    with tm.ensure_clean("test.json") as path:
-        df = DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
-        df.to_json(path, lines=True, orient="records")
-        reader = JsonReader(
-            path,
-            orient=None,
-            typ="frame",
-            dtype=True,
-            convert_axes=True,
-            convert_dates=True,
-            keep_default_dates=True,
-            precise_float=False,
-            date_unit=None,
-            encoding=None,
-            lines=True,
-            chunksize=chunksize,
-            compression=None,
-            nrows=None,
-        )
-        with reader:
-            reader.read()
-        assert reader.handles.handle.closed, (
-            f"didn't close stream with chunksize = {chunksize}"
-        )
+def test_readjson_chunks_closes(chunksize, temp_file):
+    df = DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
+    df.to_json(temp_file, lines=True, orient="records")
+    reader = JsonReader(
+        temp_file,
+        orient=None,
+        typ="frame",
+        dtype=True,
+        convert_axes=True,
+        convert_dates=True,
+        keep_default_dates=True,
+        precise_float=False,
+        date_unit=None,
+        encoding=None,
+        lines=True,
+        chunksize=chunksize,
+        compression=None,
+        nrows=None,
+    )
+    with reader:
+        reader.read()
+    assert reader.handles.handle.closed, (
+        f"didn't close stream with chunksize = {chunksize}"
+    )
 
 
 @pytest.mark.parametrize("chunksize", [0, -1, 2.2, "foo"])
@@ -278,7 +276,7 @@ def test_readjson_chunks_multiple_empty_lines(chunksize):
     tm.assert_frame_equal(orig, test, obj=f"chunksize: {chunksize}")
 
 
-def test_readjson_unicode(request, monkeypatch, engine):
+def test_readjson_unicode(request, monkeypatch, engine, temp_file):
     if engine == "pyarrow":
         # GH 48893
         reason = (
@@ -287,14 +285,13 @@ def test_readjson_unicode(request, monkeypatch, engine):
         )
         request.applymarker(pytest.mark.xfail(reason=reason, raises=ValueError))
 
-    with tm.ensure_clean("test.json") as path:
-        monkeypatch.setattr("locale.getpreferredencoding", lambda do_setlocale: "cp949")
-        with open(path, "w", encoding="utf-8") as f:
-            f.write('{"£©µÀÆÖÞßéöÿ":["АБВГДабвгд가"]}')
+    monkeypatch.setattr("locale.getpreferredencoding", lambda do_setlocale: "cp949")
+    with open(temp_file, "w", encoding="utf-8") as f:
+        f.write('{"£©µÀÆÖÞßéöÿ":["АБВГДабвгд가"]}')
 
-        result = read_json(path, engine=engine)
-        expected = DataFrame({"£©µÀÆÖÞßéöÿ": ["АБВГДабвгд가"]})
-        tm.assert_frame_equal(result, expected)
+    result = read_json(temp_file, engine=engine)
+    expected = DataFrame({"£©µÀÆÖÞßéöÿ": ["АБВГДабвгд가"]})
+    tm.assert_frame_equal(result, expected)
 
 
 @pytest.mark.parametrize("nrows", [1, 2])
@@ -441,25 +438,24 @@ def test_to_json_append_mode(mode_):
         df.to_json(mode=mode_, lines=False, orient="records")
 
 
-def test_to_json_append_output_consistent_columns():
+def test_to_json_append_output_consistent_columns(temp_file):
     # GH 35849
     # Testing that resulting output reads in as expected.
     # Testing same columns, new rows
     df1 = DataFrame({"col1": [1, 2], "col2": ["a", "b"]})
     df2 = DataFrame({"col1": [3, 4], "col2": ["c", "d"]})
 
     expected = DataFrame({"col1": [1, 2, 3, 4], "col2": ["a", "b", "c", "d"]})
-    with tm.ensure_clean("test.json") as path:
-        # Save dataframes to the same file
-        df1.to_json(path, lines=True, orient="records")
-        df2.to_json(path, mode="a", lines=True, orient="records")
+    # Save dataframes to the same file
+    df1.to_json(temp_file, lines=True, orient="records")
+    df2.to_json(temp_file, mode="a", lines=True, orient="records")
 
-        # Read path file
-        result = read_json(path, lines=True)
-        tm.assert_frame_equal(result, expected)
+    # Read path file
+    result = read_json(temp_file, lines=True)
+    tm.assert_frame_equal(result, expected)
 
 
-def test_to_json_append_output_inconsistent_columns():
+def test_to_json_append_output_inconsistent_columns(temp_file):
     # GH 35849
     # Testing that resulting output reads in as expected.
     # Testing one new column, one old column, new rows
@@ -473,17 +469,16 @@ def test_to_json_append_output_inconsistent_columns():
             "col3": [np.nan, np.nan, "!", "#"],
         }
     )
-    with tm.ensure_clean("test.json") as path:
-        # Save dataframes to the same file
-        df1.to_json(path, mode="a", lines=True, orient="records")
-        df3.to_json(path, mode="a", lines=True, orient="records")
+    # Save dataframes to the same file
+    df1.to_json(temp_file, mode="a", lines=True, orient="records")
+    df3.to_json(temp_file, mode="a", lines=True, orient="records")
 
-        # Read path file
-        result = read_json(path, lines=True)
-        tm.assert_frame_equal(result, expected)
+    # Read path file
+    result = read_json(temp_file, lines=True)
+    tm.assert_frame_equal(result, expected)
 
 
-def test_to_json_append_output_different_columns():
+def test_to_json_append_output_different_columns(temp_file):
     # GH 35849
     # Testing that resulting output reads in as expected.
     # Testing same, differing and new columns
@@ -500,19 +495,18 @@ def test_to_json_append_output_different_columns():
             "col4": [None, None, None, None, None, None, True, False],
         }
     ).astype({"col4": "float"})
-    with tm.ensure_clean("test.json") as path:
-        # Save dataframes to the same file
-        df1.to_json(path, mode="a", lines=True, orient="records")
-        df2.to_json(path, mode="a", lines=True, orient="records")
-        df3.to_json(path, mode="a", lines=True, orient="records")
-        df4.to_json(path, mode="a", lines=True, orient="records")
-
-        # Read path file
-        result = read_json(path, lines=True)
-        tm.assert_frame_equal(result, expected)
+    # Save dataframes to the same file
+    df1.to_json(temp_file, mode="a", lines=True, orient="records")
+    df2.to_json(temp_file, mode="a", lines=True, orient="records")
+    df3.to_json(temp_file, mode="a", lines=True, orient="records")
+    df4.to_json(temp_file, mode="a", lines=True, orient="records")
+
+    # Read path file
+    result = read_json(temp_file, lines=True)
+    tm.assert_frame_equal(result, expected)
 
 
-def test_to_json_append_output_different_columns_reordered():
+def test_to_json_append_output_different_columns_reordered(temp_file):
     # GH 35849
     # Testing that resulting output reads in as expected.
     # Testing specific result column order.
@@ -530,13 +524,12 @@ def test_to_json_append_output_different_columns_reordered():
             "col1": [None, None, None, None, 3, 4, 1, 2],
         }
     ).astype({"col4": "float"})
-    with tm.ensure_clean("test.json") as path:
-        # Save dataframes to the same file
-        df4.to_json(path, mode="a", lines=True, orient="records")
-        df3.to_json(path, mode="a", lines=True, orient="records")
-        df2.to_json(path, mode="a", lines=True, orient="records")
-        df1.to_json(path, mode="a", lines=True, orient="records")
-
-        # Read path file
-        result = read_json(path, lines=True)
-        tm.assert_frame_equal(result, expected)
+    # Save dataframes to the same file
+    df4.to_json(temp_file, mode="a", lines=True, orient="records")
+    df3.to_json(temp_file, mode="a", lines=True, orient="records")
+    df2.to_json(temp_file, mode="a", lines=True, orient="records")
+    df1.to_json(temp_file, mode="a", lines=True, orient="records")
+
+    # Read path file
+    result = read_json(temp_file, lines=True)
+    tm.assert_frame_equal(result, expected)