Updated the comunio test class

bierschi · May 3, 2020 · 7b9627c · 7b9627c
1 parent 77ce601
commit 7b9627c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ComunioScore/test/test_comunio.py b/ComunioScore/test/test_comunio.py
@@ -110,7 +110,8 @@ def test_get_get_community_name(self):
     def test_get_wealth(self):
 
         user_id = self.comunio.get_user_id()
-        wealth = self.comunio.get_wealth(userid=user_id)
+        wealth_data = self.comunio.get_wealth(userid=user_id)
+        wealth = wealth_data[0]
         # check if wealth is int (if no budget available, then wealth is None)
         if wealth is None:
             self.assertIsNone(wealth, msg="wealth must be of type None")