Fix number of arguments in _sum/_list functions

[hubacji1/coffee-getter.git] / ut / test_bot.py

diff --git a/ut/test_bot.py b/ut/test_bot.py
index e86613a024dde71b4717d2238ddb75b2bd1b09ae..7193c02c330dc932457791c93ef13594b5ca2956 100644 (file)
--- a/ut/test_bot.py
+++ b/ut/test_bot.py
@@ -74,23 +74,23 @@ class CoffeeGetterBotTests(TestCase):
         assert len(self.b.topeq) == 2
 
     def test_nof_args(self):
-        self.assertRaises(AssertionError, self.b.load_drunk_sum)
-        self.assertRaises(AssertionError, self.b.load_drunk_sum, ["espresso"])
+        self.assertRaises(TypeError, self.b.load_drunk_sum)
+        self.assertRaises(TypeError, self.b.load_drunk_sum, ["espresso"])
         self.assertRaises(
-            AssertionError,
+            TypeError,
             self.b.load_drunk_sum,
             ["espresso", "2010-10-20"])
         self.assertRaises(
-            AssertionError,
+            TypeError,
             self.b.load_drunk_sum,
             ["espresso", "lungo", "2010-10-20", "2020-10-20"])
-        self.assertRaises(AssertionError, self.b.load_drunk_list)
-        self.assertRaises(AssertionError, self.b.load_drunk_list, ["espresso"])
+        self.assertRaises(TypeError, self.b.load_drunk_list)
+        self.assertRaises(TypeError, self.b.load_drunk_list, ["espresso"])
         self.assertRaises(
-            AssertionError,
+            TypeError,
             self.b.load_drunk_list,
             ["espresso", "2010-10-20"])
         self.assertRaises(
-            AssertionError,
+            TypeError,
             self.b.load_drunk_list,
             ["espresso", "lungo", "2010-10-20", "2020-10-20"])