def test_remove_value_all_occurences(self): """Test remove all occurences of value""" extcsv = Writer() extcsv.add_data('TABLE', ['v1', 'v2', 'v1', 'v1', 'v2', 'v1'], field='Field1') extcsv.add_data('TABLE', ['v1', 'v2', 'v3', 'v2'], field='Field1', index=2) extcsv.remove_data('TABLE', 'Field1', 'v1', all_occurences=True) self.assertTrue('v1' not in get_data(extcsv, 'TABLE', 'Field1'), 'unexpected value found') extcsv.remove_data('TABLE', 'Field1', 'v2', index=2, all_occurences=True) self.assertFalse('v2' in get_data(extcsv, 'TABLE', 'Field1', index=2), 'unexpected value found')
def test_remove_value_by_index(self): """Test remove value by index""" extcsv = Writer() extcsv.add_data('TABLE', ['v1', 'v2', 'v1', 'v1', 'v2', 'v1'], field='Field1') extcsv.add_data('TABLE', ['v1', 'v2', 'v3', 'v2'], field='Field1', index=2) extcsv.remove_data('TABLE', 'Field1', 'v1', d_index=3) self.assertEqual('v2', get_data(extcsv, 'TABLE', 'Field1')[3], 'expected specific value') self.assertEqual(['v1', 'v2', 'v1', 'v2', 'v1'], get_data(extcsv, 'TABLE', 'Field1'), 'expected specific value') extcsv.remove_data('TABLE', 'Field1', 'v2', index=2, d_index=3) self.assertEqual('v3', get_data(extcsv, 'TABLE', 'Field1', index=2)[2], 'expected specific value')
def test_remove_value_first(self): """Test remove first occurence value""" extcsv = Writer() extcsv.add_data('TABLE', ['v1', 'v2', 'v1', 'v1', 'v2', 'v1'], field='Field1') extcsv.add_data('TABLE', ['v1', 'v2', 'v3', 'v2'], field='Field1', index=2) extcsv.remove_data('TABLE', 'Field1', 'v1') self.assertEqual('v2', get_data(extcsv, 'TABLE', 'Field1')[0], 'expected specific value') self.assertEqual(['v2', 'v1', 'v1', 'v2', 'v1'], get_data(extcsv, 'TABLE', 'Field1'), 'expected specific value') extcsv.remove_data('TABLE', 'Field1', 'v2', index=2) self.assertEqual('v1', get_data(extcsv, 'TABLE', 'Field1', index=2)[0], 'expected specific value')