1 from StringIO import StringIO

     2 from logilab.common.testlib import TestCase, unittest_main

     3 from cubicweb import dataimport

     4 class UcsvreaderTC(TestCase):

     5 

     6     def test_empty_lines_skipped(self):

     7         stream = StringIO('''a,b,c,d,

     8 1,2,3,4,

     9 ,,,,

    10 ,,,,

    11 ''')

    12         self.assertEqual([[u'a', u'b', u'c', u'd', u''],

    13                           [u'1', u'2', u'3', u'4', u''],

    14                           ],

    15                          list(dataimport.ucsvreader(stream)))

    16         stream.seek(0)

    17         self.assertEqual([[u'a', u'b', u'c', u'd', u''],

    18                           [u'1', u'2', u'3', u'4', u''],

    19                           [u'', u'', u'', u'', u''],

    20                           [u'', u'', u'', u'', u'']

    21                           ],

    22                          list(dataimport.ucsvreader(stream, skip_empty=False)))

    23 

    24 

    25 if __name__ == '__main__':

    26     unittest_main()