def testStringSource_one_col_04(self):
from datagraph.datasources import StringSource
data = '''
123
234
4560
1000
'''
source = StringSource(data)
items = list(source.getRowsIterator())
self.assertEqual(len(items), 4)
self.assertEqual(items[0], ['123'])
self.assertEqual(items[1], ['234'])
self.assertEqual(items[2], ['4560'])
self.assertEqual(items[3], ['1000'])
def testStringSource_col_02(self):
from datagraph.datasources import StringSource
data = u'''
123 456 789
234 100 111
456 101 99
-10\t55 66
20 30 40
'''
source = StringSource(data)
items = list(source.getRowsIterator())
self.assertEqual(len(items), 5)
self.assertEqual(items[0], [u'123', u'456', u'789'])
self.assertEqual(items[1], [u'234', u'100', u'111'])
self.assertEqual(items[2], [u'456', u'101', u'99'])
self.assertEqual(items[3], [u'-10', u'55', u'66'])
self.assertEqual(items[4], [u'20', u'30', u'40'])
