From 0d483234d4a364e1f5b77bbb37558bad5d76fb31 Mon Sep 17 00:00:00 2001
From: hugovk <hugovk@users.noreply.github.com>
Date: Sat, 1 Mar 2014 16:28:40 +0200
Subject: [PATCH] Add test case for bug already fixed by @inversion in
 https://github.com/hugovk/pylast/commit/c8216a139e216f016773de3eb92a1044ea919c1e.
 Closes #64.

---
 test_pylast.py | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/test_pylast.py b/test_pylast.py
index f18037d..073e3be 100755
--- a/test_pylast.py
+++ b/test_pylast.py
@@ -14,7 +14,7 @@ def load_secrets():
         doc = yaml.load(f)
     return doc
 
-class TestSequenceFunctions(unittest.TestCase):
+class TestPyLast(unittest.TestCase):
 
     secrets = None
 
@@ -157,6 +157,7 @@ class TestSequenceFunctions(unittest.TestCase):
         self.assertEqual(str(loved[0].track.artist), "Test Artist")
         self.assertEqual(str(loved[0].track.title), "Test Title")
 
+
     def test_unlove(self):
         # Arrange
         artist = "Test Artist"
@@ -175,7 +176,34 @@ class TestSequenceFunctions(unittest.TestCase):
             self.assertNotEqual(str(loved.track.title), "Test Title")
 
 
+    def test_get_100_albums(self):
+        # Arrange
+        library = pylast.Library(user = self.username, network = self.network)
+
+        # Act
+        albums = library.get_albums(limit = 100)
+
+        # Assert
+        self.assertGreaterEqual(len(albums), 0)
+
+
+    def test_get_limitless_albums(self):
+        # Arrange
+        library = pylast.Library(user = self.username, network = self.network)
+
+        # Act
+        albums = library.get_albums(limit = None)
+
+        # Assert
+        self.assertGreaterEqual(len(albums), 0)
+
+
 if __name__ == '__main__':
+
+#     suite = unittest.TestSuite()
+#     suite.addTest(TestPyLast('test_get_limitless_albums'))
+#     unittest.TextTestRunner().run(suite)
+
     unittest.main()
 
 # End of file