def test_in(self):
"""Test in operator."""
store = LabelStore(['laBel1', 'label2', 'Label3'])
self.assertTrue('label1' in store)
self.assertTrue('laBel2' in store)
self.assertTrue('labeL3' in store)
self.assertFalse('label' in store)
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_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)
def test_add(self):
"""Test adding items."""
store = LabelStore()
store.add('laBel1')
self.assertItemsEqual(['laBel1'], store)
self.assertItemsEqual(['laBel1'], store.added)
self.assertItemsEqual([], store.removed)
store.add('label2')
self.assertItemsEqual(['laBel1', 'label2'], store)
self.assertItemsEqual(['laBel1', 'label2'], store.added)
self.assertItemsEqual([], store.removed)
store.add('labEl2')
self.assertItemsEqual(['laBel1', 'labEl2'], store)
self.assertItemsEqual(['laBel1', 'labEl2'], store.added)
self.assertItemsEqual([], store.removed)
def test_init_and_iter(self): """Test initializing and iterating LabelStore.""" store = LabelStore(['label1', 'label2']) self.assertItemsEqual([], store.added) self.assertItemsEqual([], store.removed) self.assertItemsEqual(['label1', 'label2'], list(store))
def test_remove_by_prefix(self):
"""Test remove_by_prefix."""
store = LabelStore(['p-0', 'P-1', 'Q-2'])
store.remove_by_prefix('p-')
self.assertItemsEqual(['Q-2'], store)