Add test case for Artist.get_correction()

This commit is contained in:
José Neto 2015-07-01 17:04:01 -03:00
parent db9bc4f216
commit 6b58f43c01

View file

@ -1907,7 +1907,16 @@ class TestPyLast(unittest.TestCase):
self.assertEqual(len(tracks), 1)
self.assertEqual(str(tracks[0].track.artist), "Johnny Cash")
self.assertEqual(str(tracks[0].track.title), "Ring of Fire")
def test_artist_get_correction(self):
# Arrange
artist = pylast.Artist("guns and roses", self.network)
# Act
corrected_artist_name = artist.get_correction()
# Assert
self.assertEqual(corrected_artist_name, "Guns N' Roses")
if __name__ == '__main__':
unittest.main(failfast=True)