def test_remove_album_working(self): band = Band("Death") album = Album("The Sound of Perseverance") band.add_album(album) message = band.remove_album("The Sound of Perseverance") expected = "Album The Sound of Perseverance has been removed." self.assertEqual(message, expected)
def test_add_album_already_added(self): band = Band("Death") album = Album("The Sound of Perseverance") band.add_album(album) message = band.add_album(album) expected = "Band Death already has The Sound of Perseverance in their library." self.assertEqual(message, expected)
def test_remove_album_published(self): band = Band("Death") album = Album("The Sound of Perseverance") album.publish() band.add_album(album) message = band.remove_album("The Sound of Perseverance") expected = "Album has been published. It cannot be removed." self.assertEqual(message, expected)
from project.band import Band from project.album import Album from project.song import Song if __name__ == "__main__": song = Song("Running in the 90s", 3.45, False) print(song.get_info()) album = Album("Initial D", song) second_song = Song("Around the World", 2.34, False) print(album.add_song(second_song)) print(album.details()) print(album.publish()) band = Band("Manuel") print(band.add_album(album)) print(band.remove_album("Initial D")) print(band.details())
def test_add_album_working(self): band = Band("Death") album = Album("The Sound of Perseverance") message = band.add_album(album) expected = "Band Death has added their newest album The Sound of Perseverance." self.assertEqual(message, expected)
def test_init(self): band = Band("Death") message = f"{band.name} - {len(band.albums)}" expected = "Death - 0" self.assertEqual(message, expected)
def test_details(self): band = Band("Death") message = band.details() expected = "Band Death\n" self.assertEqual(message, expected)
def test_remove_album_not_found(self): band = Band("Death") album = Album("The Sound of Perseverance") message = band.remove_album("The Sound of Perseverance") expected = "Album The Sound of Perseverance is not found." self.assertEqual(message, expected)
from project.song import Song from project.album import Album from project.band import Band song = Song("Running in the 90s", 3.45, False) song_1 = Song("Song Two", 3.45, False) song_2 = Song("Song Three", 3.45, False) song_3 = Song("Song Four", 3.45, False) print(song.get_info()) album1 = Album("Initial D", song, song_1) album2 = Album("Album Two", song_2, song_3) band = Band("Manuel") print(band.add_album(album1)) print(band.remove_album("Album Two")) # print(band.add_album(album2)) print(band.details())