Parametrize test

Author: kinow

pandas/tests/io/formats/test_to_csv.py

@@ -518,10 +518,12 @@ def test_to_csv_compression(self, compression_only, read_infer, to_infer):
             result = pd.read_csv(path, index_col=0, compression=read_compression)
             tm.assert_frame_equal(result, df)
 
-    def test_to_csv_na_rep_long_string(self):
+    @pytest.mark.parametrize("df_value", [float('nan')])
+    @pytest.mark.parametrize("df_new_type", ["Int64"])
+    def test_to_csv_na_rep_long_string(self, df_value, df_new_type):
         # see gh-25099
-        df = pd.DataFrame({"c": [float('nan')] * 3})
-        df = df.astype("Int64")
+        df = pd.DataFrame({"c": [df_value] * 3})
+        df = df.astype(df_new_type)
         expected_rows = ['c',
                          'mynull',
                          'mynull',