def test_add_field_1(self):
"""Test adding new field to table"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field not found')
def test_add_field_1(self):
"""Test adding new field to table"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field not found')
def test_add_field_3(self):
"""Test order of insert of fields"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
keys = extcsv.extcsv_ds['CONTENT$1'].keys()
self.assertEqual(0, keys.index('comments'), 'index 0 mismatch')
self.assertEqual(1, keys.index('Class'), 'index 1 mismatch')
self.assertEqual(2, keys.index('Category'), 'index 2 mismatch')
self.assertEqual(3, keys.index('Level'), 'index 3 mismatch')
def test_add_field_3(self):
"""Test order of insert of fields"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
keys = extcsv.extcsv_ds['CONTENT$1'].keys()
self.assertEqual(0, keys.index('comments'), 'index 0 mismatch')
self.assertEqual(1, keys.index('Class'), 'index 1 mismatch')
self.assertEqual(2, keys.index('Category'), 'index 2 mismatch')
self.assertEqual(3, keys.index('Level'), 'index 3 mismatch')
def test_add_field_2(self):
"""Test adding multiple new fields to table"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table CONTENT not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Category' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Level' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
def test_add_field_2(self):
"""Test adding multiple new fields to table"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table CONTENT not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Category' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Level' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
def test_add_value_3(self):
"""Test adding value to table given identical table names"""
extcsv = Writer()
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time')
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time', index=2)
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time', index=3)
extcsv.add_data('GLOBAL', '290.0', index=2, field='Wavelength')
extcsv.add_data('GLOBAL', '07:28:49', index=3, field='Time')
extcsv.add_data('GLOBAL', '1.700E-06', field='S-Irradiance')
self.assertEqual(['290.0'],
get_data(extcsv, 'GLOBAL', 'Wavelength', index=2),
'expected specific value')
self.assertEqual(['07:28:49'],
get_data(extcsv, 'GLOBAL', 'Time', index=3),
'expected specific value')
self.assertEqual(['1.700E-06'],
get_data(extcsv, 'GLOBAL', 'S-Irradiance'),
'expected specific value')
def test_add_value_3(self):
"""Test adding value to table given identical table names"""
extcsv = Writer()
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time')
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time', index=2)
extcsv.add_field('GLOBAL', 'Wavelength,S-Irradiance,Time', index=3)
extcsv.add_data('GLOBAL', '290.0', index=2, field='Wavelength')
extcsv.add_data('GLOBAL', '07:28:49', index=3, field='Time')
extcsv.add_data('GLOBAL', '1.700E-06', field='S-Irradiance')
self.assertEqual(['290.0'],
get_data(extcsv, 'GLOBAL', 'Wavelength', index=2),
'expected specific value')
self.assertEqual(['07:28:49'],
get_data(extcsv, 'GLOBAL', 'Time', index=3),
'expected specific value')
self.assertEqual(['1.700E-06'],
get_data(extcsv, 'GLOBAL', 'S-Irradiance'),
'expected specific value')
def test_add_value_2(self):
"""Test adding new value to existing table.field, horizontally"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
extcsv.add_data('CONTENT', 'a,b,c')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table CONTENT not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Category' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Level' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('a' in get_data(extcsv, 'CONTENT', 'Class'),
'value CONTENT.Class not found')
self.assertTrue('b' in get_data(extcsv, 'CONTENT', 'Category'),
'value CONTENT.Class not found')
self.assertTrue('c' in get_data(extcsv, 'CONTENT', 'Level'),
'value CONTENT.Class not found')
def test_add_value_2(self):
"""Test adding new value to existing table.field, horizontally"""
extcsv = Writer()
extcsv.add_field('CONTENT', 'Class,Category,Level')
extcsv.add_data('CONTENT', 'a,b,c')
self.assertTrue('CONTENT$1' in extcsv.extcsv_ds.keys(),
'table CONTENT not found')
self.assertTrue('Class' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Category' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('Level' in extcsv.extcsv_ds['CONTENT$1'].keys(),
'field CONTENT.Class not found')
self.assertTrue('a' in get_data(extcsv, 'CONTENT', 'Class'),
'value CONTENT.Class not found')
self.assertTrue('b' in get_data(extcsv, 'CONTENT', 'Category'),
'value CONTENT.Class not found')
self.assertTrue('c' in get_data(extcsv, 'CONTENT', 'Level'),
'value CONTENT.Class not found')
