def test_remove_many(self):
pi = PictureIndex(self.index)
mock_pics = [mock.Mock(), mock.Mock(), mock.Mock()]
pi.add(mock_pics)
pi.remove(mock_pics[:2])
self.assertNotIn(mock_pics[0], pi.pics())
self.assertNotIn(mock_pics[1], pi.pics())
self.assertIn(mock_pics[2], pi.pics())
def test_pics(self):
pi = PictureIndex(self.index)
self.assertSequenceEqual(pi.pics(), sorted(self.index.values()))
def test_add_many(self):
pi = PictureIndex(self.index)
mock_pics = [mock.Mock(), mock.Mock(), mock.Mock()]
pi.add(mock_pics)
for mock_pic in mock_pics:
self.assertIn(mock_pic, pi.pics())
