diff --git a/scaruffi/tests.py b/scaruffi/tests.py new file mode 100644 index 0000000..b3fa9e7 --- /dev/null +++ b/scaruffi/tests.py @@ -0,0 +1,21 @@ +import unittest + +from scaruffi import api + + +class TestScaruffi(unittest.TestCase): + + def setUpClass(): + api.setup_logging("test") + + def test_get_musicians(self): + musicians = api.get_musicians() + self.assertEqual(len(musicians), 20) + + def test_get_ratings(self): + self.assertIsNotNone(api.get_ratings(1960)) + self.assertIsNotNone(api.get_ratings(1970)) + self.assertIsNotNone(api.get_ratings(1980)) + self.assertIsNotNone(api.get_ratings(1990)) + self.assertIsNotNone(api.get_ratings(2000)) + self.assertIsNotNone(api.get_ratings(2010))