def test_read_write_csv_dialect(self): expected = [['this is some text', 'scandal', '42'], ["here's some more text: boom!", 'escándalo', '1.0']] filename = os.path.join( self.tempdir, 'test_read_write_csv.csv') fileio.write_csv( expected, filename, dialect='excel', auto_make_dirs=True) observed = list(fileio.read_csv(filename, dialect='infer')) self.assertEqual(observed, expected)
def test_read_write_csv_delimiters(self): expected = [['this is some text', 'scandal', '42'], ["here's some more text: boom!", 'escándalo', '1.0']] for delimiter in (',', '\t', '|', ':'): filename = os.path.join( self.tempdir, 'test_read_write_csv.csv') fileio.write_csv( expected, filename, delimiter=delimiter, auto_make_dirs=True) observed = list(fileio.read_csv(filename, delimiter=delimiter)) self.assertEqual(observed, expected)
def test_read_write_csv_compressed(self): expected = [['this is some text', 'scandal', '42'], ["here's some more text: boom!", 'escándalo', '1.0']] for ext in ('.csv', '.csv.gz', '.csv.bz2', '.csv.xz'): filename = os.path.join(self.tempdir, 'test_read_write_csv' + ext) if is_python2 is True and ext != '.csv': self.assertRaises(ValueError, fileio.open_sesame, filename, 'wt', None, True) else: fileio.write_csv(expected, filename, auto_make_dirs=True) observed = list(fileio.read_csv(filename)) self.assertEqual(observed, expected)
def test_read_write_csv_compressed(self): expected = [['this is some text', 'scandal', '42'], ["here's some more text: boom!", 'escándalo', '1.0']] for ext in ('.csv', '.csv.gz', '.csv.bz2', '.csv.xz'): filename = os.path.join( self.tempdir, 'test_read_write_csv' + ext) if PY2 is True and ext != '.csv': self.assertRaises( ValueError, fileio.open_sesame, filename, 'wt', None, True) else: fileio.write_csv(expected, filename, auto_make_dirs=True) observed = list(fileio.read_csv(filename)) self.assertEqual(observed, expected)