diff --git a/pylast.py b/pylast.py index 63c1fbb..0c40b49 100644 --- a/pylast.py +++ b/pylast.py @@ -2039,6 +2039,17 @@ class Library(_BaseObject): self._request("library.addArtist", False, params) + def remove_artist(self, artist): + """Remove an artist from this library.""" + + params = self._get_params() + if type(artist) == str: + params["artist"] = artist + else: + params["artist"] = artist.get_name() + + self._request("library.removeArtist", False, params) + def add_track(self, track): """Add a track to this library.""" diff --git a/test_pylast.py b/test_pylast.py index b07da94..9655242 100755 --- a/test_pylast.py +++ b/test_pylast.py @@ -121,6 +121,25 @@ class TestPyLast(unittest.TestCase): self.assertTrue(value) + def test_remove_artist(self): + # Arrange + artist = "Test Artist 2" + library = pylast.Library(user = self.username, network = self.network) + library.add_artist(artist) + + # Act + library.remove_artist(artist) + + # Assert + artists = library.get_artists() + for artist in artists: + value = (str(artist[0]) == "Test Artist 2") + if value: + break + self.assertFalse(value) + + + def test_get_venue(self): # Arrange venue_name = "Last.fm Office" country_name = "United Kingom" @@ -305,6 +324,7 @@ if __name__ == '__main__': # suite = unittest.TestSuite() # suite.addTest(TestPyLast('test_add_artist')) +# suite.addTest(TestPyLast('test_remove_artist')) # unittest.TextTestRunner().run(suite) unittest.main()