Oops, left in some typings

Eldinnie · Eldinnie · commit c5f703a67ff0 · 2018-02-22T11:23:52.000+01:00
diff --git a/tests/test_basepersistence.py b/tests/test_basepersistence.py
@@ -112,7 +112,7 @@ def get_chat_data():
         u.dispatcher.chat_data[442233]['test5'] = 'test6'
         assert u.dispatcher.chat_data[442233]['test5'] == 'test6'
 
-    def test_dispatcher_integration_handlers(self, caplog, bot, base_persistence: BasePersistence,
+    def test_dispatcher_integration_handlers(self, caplog, bot, base_persistence,
                                              chat_data, user_data):
         def get_user_data():
             return user_data
diff --git a/tests/test_pickle_persistence.py b/tests/test_pickle_persistence.py
@@ -160,7 +160,7 @@ def test_with_good_single_file(self, pickle_persistence, good_pickle_files):
         with pytest.raises(KeyError):
             conversation2[(123, 123)]
 
-    def test_updating_multi_file(self, pickle_persistence: PicklePersistence, good_pickle_files):
+    def test_updating_multi_file(self, pickle_persistence, good_pickle_files):
         user_data = pickle_persistence.get_user_data()
         user_data[54321]['test9'] = 'test 10'
         assert not pickle_persistence.user_data == user_data
@@ -188,7 +188,7 @@ def test_updating_multi_file(self, pickle_persistence: PicklePersistence, good_p
             conversations_test = defaultdict(dict, pickle.load(f))
         assert conversations_test['name1'] == conversation1
 
-    def test_updating_single_file(self, pickle_persistence: PicklePersistence, good_pickle_files):
+    def test_updating_single_file(self, pickle_persistence, good_pickle_files):
         pickle_persistence.single_file = True
 
         user_data = pickle_persistence.get_user_data()
