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_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_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')