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()) == []
