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