TST: Add to_csv test for writing the single column CSV

Author: Licht-T

pandas/tests/io/formats/test_to_csv.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import sys
 import numpy as np
 import pandas as pd
 import pytest
@@ -9,6 +10,34 @@
 
 class TestToCSV(object):
 
+    @pytest.mark.xfail((3, 6, 5) > sys.version_info >= (3, 5),
+                       reason='Python csv library bug')
+    def test_to_csv_with_single_column(self):
+        # GH18676
+        df1 = DataFrame([None, 1],
+                        header=None,
+                        index=None)
+        expected1 = """\
+""
+1
+"""
+        with tm.ensure_clean('test.csv') as path:
+            df1.to_csv(path, header=None, index=None)
+            with open(path, 'r') as f:
+                assert f.read() == expected1
+
+        df2 = DataFrame([1, None],
+                        header=None,
+                        index=None)
+        expected2 = """\
+1
+""
+"""
+        with tm.ensure_clean('test.csv') as path:
+            df2.to_csv(path, header=None, index=None)
+            with open(path, 'r') as f:
+                assert f.read() == expected2
+
     def test_to_csv_defualt_encoding(self):
         # GH17097
         df = DataFrame({'col': [u"AAAAA", u"ÄÄÄÄÄ", u"ßßßßß", u"聞聞聞聞聞"]})