diff --git a/contrib_bots/bots/define/test_define.py b/contrib_bots/bots/define/test_define.py index 770f1fd..fc67236 100644 --- a/contrib_bots/bots/define/test_define.py +++ b/contrib_bots/bots/define/test_define.py @@ -42,7 +42,7 @@ class TestDefineBot(TestCase): messages.append(message2) return messages - def test_define(self): + def runTest(self): # type: None -> None # Edit bot_module to test different bots, the below code can be looped for all the bots. bot_module = os.path.join(our_dir, "define.py") diff --git a/contrib_bots/bots_test_lib.py b/contrib_bots/bots_test_lib.py index 3d17463..14c5186 100644 --- a/contrib_bots/bots_test_lib.py +++ b/contrib_bots/bots_test_lib.py @@ -8,14 +8,13 @@ import sys import unittest from mock import MagicMock, patch -from unittest import TestCase from run import get_lib_module from bot_lib import StateHandler from contrib_bots import bot_lib from six.moves import zip -class BotTestCase(TestCase): +class BotTestCase(): def mock_test(self, messages, message_handler, bot_response): # type: (List[Dict[str, str]], Function) -> None