class SetOperationTest(unittest.TestCase): def setUp(self): self.song = Mock() self.tags = {'one': 1, 'two': 2, 'three': 3} self.set_operation = SetOperation(self.tags) def test_apply(self): self.set_operation.apply(self.song) self.song.update.assert_called_with(self.tags)