def test_reset(self):
"""Test reset."""
store = LabelStore(['laBel1', 'label2', 'Label3'])
store.add('label4')
store.add('label5')
store.remove('label1')
store.reset()
self.assertItemsEqual([], store.added)
self.assertItemsEqual([], store.removed)
def test_reset(self):
"""Test reset."""
store = LabelStore(['laBel1', 'label2', 'Label3'])
store.add('label4')
store.add('label5')
store.remove('label1')
store.reset_tracking()
six.assertCountEqual(self, [], store.added)
six.assertCountEqual(self, [], store.removed)
def test_add_and_remove(self):
"""Test both adding and removing."""
store = LabelStore(['laBel1', 'label2', 'Label3'])
store.remove('Label1')
self.assertItemsEqual(['label2', 'Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual(['Label1'], store.removed)
store.add('label1')
self.assertItemsEqual(['label1', 'label2', 'Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual([], store.removed)
store.remove('Label1')
store.add('label4')
self.assertItemsEqual(['label2', 'Label3', 'label4'], store)
self.assertItemsEqual(['label4'], store.added)
self.assertItemsEqual(['Label1'], store.removed)
def test_remove(self):
"""Test removing items."""
store = LabelStore(['laBel1', 'label2', 'Label3'])
store.remove('Label1')
self.assertItemsEqual(['label2', 'Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual(['Label1'], store.removed)
store.remove('Label2')
self.assertItemsEqual(['Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual(['Label1', 'Label2'], store.removed)
store.remove('LaBel2')
self.assertItemsEqual(['Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual(['Label1', 'Label2'], store.removed)
store.remove('Label4')
self.assertItemsEqual(['Label3'], store)
self.assertItemsEqual([], store.added)
self.assertItemsEqual(['Label1', 'Label2'], store.removed)
