def test_writes_id(self):
fieldnames = ['id', 'a[]']
w = NestedDictWriter(self.f, fieldnames)
w.writerows([{
'a': [1, 2]
}, {
'a': [3]
}, {
'a': [4, 5, 6],
'id': 200
}],
first_id=100)
self.assertEqual(self.read_all(), [
['100', '1'],
['100', '2'],
['101', '3'],
['200', '4'],
['200', '5'],
['200', '6'],
])
def test_root(self):
fieldnames = ['']
w = NestedDictWriter(self.f, fieldnames)
w.writerows([1, 2, 3, 4])
self.assertEqual(self.read_all(), [['1'], ['2'], ['3'], ['4']])