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)