diff --git a/test/test_replaygain.py b/test/test_replaygain.py index 194e83688..548220c0f 100644 --- a/test/test_replaygain.py +++ b/test/test_replaygain.py @@ -41,6 +41,8 @@ def reset_replaygain(item): item['rg_track_gain'] = None item['rg_album_gain'] = None item['rg_album_gain'] = None + item['r128_track_gain'] = None + item['r128_album_gain'] = None item.write() item.store() @@ -64,16 +66,6 @@ class ReplayGainCliTestBase(TestHelper): self.teardown_beets() self.unload_plugins() - def _reset_replaygain(self, item): - item['rg_track_peak'] = None - item['rg_track_gain'] = None - item['rg_album_peak'] = None - item['rg_album_gain'] = None - item['r128_track_gain'] = None - item['r128_album_gain'] = None - item.write() - item.store() - def test_cli_saves_track_gain(self): for item in self.lib.items(): self.assertIsNone(item.rg_track_peak) @@ -137,7 +129,7 @@ class ReplayGainCliTestBase(TestHelper): album = self.add_album_fixture(2, ext="opus") for item in album.items(): - self._reset_replaygain(item) + reset_replaygain(item) self.run_command('replaygain', '-a') @@ -155,7 +147,7 @@ class ReplayGainCliTestBase(TestHelper): def analyse(target_level): self.config['replaygain']['targetlevel'] = target_level - self._reset_replaygain(item) + reset_replaygain(item) self.run_command('replaygain', '-f') mediafile = MediaFile(item.path) return mediafile.rg_track_gain