diff --git a/tests/base_case.py b/tests/base_case.py index fca2f63fa..1f128bf1d 100644 --- a/tests/base_case.py +++ b/tests/base_case.py @@ -31,13 +31,6 @@ def get_kwargs(self): 'database': None } - def random_string(self, start=0, end=9000): - """ - Generate a string based on a random number. - """ - from random import randint - return str(randint(start, end)) - class ChatBotMongoTestCase(ChatBotTestCase): @@ -58,7 +51,7 @@ def setUpClass(cls): def get_kwargs(self): kwargs = super(ChatBotMongoTestCase, self).get_kwargs() - kwargs['database'] = self.random_string() + kwargs['database'] = 'chatterbot_test_database' kwargs['storage_adapter'] = 'chatterbot.storage.MongoDatabaseAdapter' return kwargs