class TestPlaylist(unittest.TestCase): """docstring for TestPlaylist""" def setUp(self): self.p = Playlist(name="Code", repeat=False, shuffle=False) self.song_1 = Song(title="Odin", artist="Manowar", album="The Sons of Odin", length="60") self.song_2 = Song(title="Odin2", artist="Manowar2", album="The Sons of Odin2", length="60") def test_Playlist_addsong(self): self.p.add_song(self.song_1) self.assertIn(self.song_1, self.p.playlist) def test_Playlist_addsongs(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.assertIn(self.song_1, self.p.playlist) self.assertIn(self.song_2, self.p.playlist) self.assertEqual(len(self.p.playlist), 2) def test_Playlist_remove_song(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.assertEqual(self.p.remove_song(self.song_1), self.song_1) self.assertEqual(len(self.p.playlist), 1) #tank_size - how much our car can ride with a full tank of gas. It is a positive integer in kilometers def test_Playlist_total_length(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.assertEqual(self.p.total_length(), "0:02:00") def test_Playlist_artists(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.p.add_song(self.song_1) def test_Playlist_shuffle(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.p._shuffle = True self.assertNotIn(self.p.shuffle(), self.p._shuffle_playlist) self.assertNotIn(self.p.shuffle(), self.p._shuffle_playlist) #self.assertEqual(self.p.shuffle(), 0) def test_Playlist_next_song(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) def test_Playlist_pprint_playlist(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) print(self.p.pprint_playlist()) def test_Playlist_save(self): songs = [self.song_1, self.song_2] self.p.add_songs(songs) self.p.save('export.json') def test_Playlist_load(self): code = Playlist.load('export.json')