def test_artists_with_multiple_songs(self): artist = Artist('artist name', 'country') first_album = Album('first album title', 2020) first_album.add(Song('first song title')) first_album.add(Song('second song title')) second_album = Album('second album title', 2020) second_album.add(Song('third song title')) artist.add_album(first_album) artist.add_album(second_album) expected_songs = [ Song('first song title'), Song('second song title'), Song('third song title') ] assert list(artist.get_songs()) == expected_songs
def test_artist_with_one_song(self): artist = Artist('artist name', 'country') album = Album('album title', 2020) album.add(Song('song title')) artist.add_album(album) assert list(artist.get_songs()) == [Song('song title')]
def test_artist_with_albums_but_no_songs(self): artist = Artist('artist name', 'country') artist.add_album(Album('album title', 2020)) assert list(artist.get_songs()) == []