pre-commit checks

Author: eicchen

doc/source/whatsnew/v3.0.0.rst

@@ -773,9 +773,9 @@ I/O
 - Bug in :meth:`DataFrame.to_excel` when writing empty :class:`DataFrame` with :class:`MultiIndex` on both axes (:issue:`57696`)
 - Bug in :meth:`DataFrame.to_excel` where the :class:`MultiIndex` index with a period level was not a date (:issue:`60099`)
 - Bug in :meth:`DataFrame.to_stata` when exporting a column containing both long strings (Stata strL) and :class:`pd.NA` values (:issue:`23633`)
+- Bug in :meth:`DataFrame.to_stata` when input encoded length and normal length are mismatched (:issue:`61583`)
 - Bug in :meth:`DataFrame.to_stata` when writing :class:`DataFrame` and ``byteorder=`big```. (:issue:`58969`)
 - Bug in :meth:`DataFrame.to_stata` when writing more than 32,000 value labels. (:issue:`60107`)
-- Bug in :meth:`DataFrame.to_stata` when input encoded length and normal length are mismatched (:issue:`61583`)
 - Bug in :meth:`DataFrame.to_string` that raised ``StopIteration`` with nested DataFrames. (:issue:`16098`)
 - Bug in :meth:`HDFStore.get` was failing to save data of dtype datetime64[s] correctly (:issue:`59004`)
 - Bug in :meth:`HDFStore.select` causing queries on categorical string columns to return unexpected results (:issue:`57608`)

pandas/io/stata.py

@@ -3263,11 +3263,11 @@ def generate_blob(self, gso_table: dict[str, tuple[int, int]]) -> bytes:
             bio.write(gso_type)
 
             # llll
-            if(isinstance(strl, str)):
+            if isinstance(strl, str):
                 strl_convert = bytes(strl, "utf-8")
             else:
                 strl_convert = strl
-                
+
             bio.write(struct.pack(len_type, len(strl_convert) + 1))
 
             # xxx...xxx

pandas/tests/io/test_stata.py

@@ -2602,11 +2602,12 @@ def test_strl_missings(temp_file, version):
     )
     df.to_stata(temp_file, version=version)
 
+
 @pytest.mark.parametrize("version", [117, 118, 119, None])
 def test_ascii_error(temp_file, version):
     # GH #61583
     # Check that 2 byte long unicode characters doesn't cause export error
-    df = pd.DataFrame({'doubleByteCol': ['§'*1500]})
+    df = DataFrame({"doubleByteCol": ["§" * 1500]})
     df.to_stata(temp_file, write_index=0, version=version)
     df_input = read_stata(temp_file)
-    tm.assert_frame_equal(df, df_input)
+    tm.assert_frame_equal(df, df_input)