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)