From 22e03705c496ca5e7febe28d7cbc002c91ab2d0f Mon Sep 17 00:00:00 2001
From: Hugo <hugovk@users.noreply.github.com>
Date: Thu, 11 Jan 2018 10:37:01 +0200
Subject: [PATCH] Passing test for limit=1 or 50, failing test for 100

---
 tests/test_pylast_artist.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/tests/test_pylast_artist.py b/tests/test_pylast_artist.py
index 66dbb49..658f1d3 100755
--- a/tests/test_pylast_artist.py
+++ b/tests/test_pylast_artist.py
@@ -86,6 +86,42 @@ class TestPyLastArtist(PyLastTestCase):
         # Assert
         self.helper_two_different_things_in_top_list(things, pylast.Album)
 
+    def test_artist_top_albums_limit_1(self):
+        # Arrange
+        limit = 1
+        # Pick an artist with plenty of plays
+        artist = self.network.get_top_artists(limit=1)[0].item
+
+        # Act
+        things = artist.get_top_albums(limit=limit)
+
+        # Assert
+        self.assertEqual(len(things), 1)
+
+    def test_artist_top_albums_limit_50(self):
+        # Arrange
+        limit = 50
+        # Pick an artist with plenty of plays
+        artist = self.network.get_top_artists(limit=1)[0].item
+
+        # Act
+        things = artist.get_top_albums(limit=limit)
+
+        # Assert
+        self.assertEqual(len(things), 50)
+
+    def test_artist_top_albums_limit_100(self):
+        # Arrange
+        limit = 100
+        # Pick an artist with plenty of plays
+        artist = self.network.get_top_artists(limit=1)[0].item
+
+        # Act
+        things = artist.get_top_albums(limit=limit)
+
+        # Assert
+        self.assertEqual(len(things), 100)
+
     def test_artist_listener_count(self):
         # Arrange
         artist = self.network.get_artist("Test Artist")