curds: Remove extra Root lookup step from these function calls

We still do this internally in the playlist/__init__.py file, so we can
skip it here and make the lines a little shorter.

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
Anna Schumaker 2019-12-14 08:19:30 -05:00
parent f5262278df
commit 38c6e9d35f
1 changed files with 6 additions and 6 deletions

View File

@ -25,18 +25,18 @@ class TestPlaylist(unittest.TestCase):
self.assertEqual(playlist.lookup("Playlists"), playlist.Root.lookup("Playlists"))
def test_starred_shortcut(self):
self.assertEqual(playlist.Starred, playlist.Root.lookup("Playlists").lookup("Starred"))
self.assertEqual(playlist.Starred, playlist.lookup("Playlists").lookup("Starred"))
playlist.Starred = None
playlist.reset()
self.assertEqual(playlist.Starred, playlist.Root.lookup("Playlists").lookup("Starred"))
self.assertEqual(playlist.Starred, playlist.lookup("Playlists").lookup("Starred"))
def test_up_next_shortcut(self):
self.assertEqual(playlist.UpNext, playlist.Root.lookup("Up Next"))
self.assertEqual(playlist.UpNext, playlist.lookup("Up Next"))
playlist.UpNext = None
playlist.reset()
self.assertEqual(playlist.UpNext, playlist.Root.lookup("Up Next"))
self.assertEqual(playlist.UpNext, playlist.lookup("Up Next"))
def test_save_playlists(self):
dfile = data.DataFile("playlists.pickle", data.READ)
@ -56,5 +56,5 @@ class TestPlaylist(unittest.TestCase):
playlist.load()
self.assertNotEqual(playlist.Root, root)
self.assertEqual(playlist.Starred, playlist.Root.lookup("Playlists").lookup("Starred"))
self.assertEqual(playlist.UpNext, playlist.Root.lookup("Up Next"))
self.assertEqual(playlist.Starred, playlist.lookup("Playlists").lookup("Starred"))
self.assertEqual(playlist.UpNext, playlist.lookup("Up Next"))