def test_to_and_from_str_with_albums(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', 2021)
second_album.add(Song('third song title'))
assert Artist.from_string(str(artist)) == artist
def test_to_and_from_str_with_albums_without_songs(self):
artist = Artist('artist name', 'country')
artist.add_album(Album('first album name', 2020))
artist.add_album(Album('second album name', 2020))
assert Artist.from_string(str(artist)) == artist
def test_to_and_from_str_without_albums(self):
artist = Artist('artist name', 'country')
assert Artist.from_string(str(artist)) == artist
