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_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_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())