def generateDataBase(self):
from database import InnerVoiceDataBase
database = InnerVoiceDataBase()
for file_path in self._file_list:
contents = self._readFileContents(file_path)
for line in contents.split("\n"):
if not line.strip() == "":
database.add(line.strip())
return database
def setUp(self):
from database import InnerVoiceDataBase
self.database = InnerVoiceDataBase()
class InnerVoiceDataBaseTest(unittest.TestCase):
def setUp(self):
from database import InnerVoiceDataBase
self.database = InnerVoiceDataBase()
def test_add(self):
self.database.add("aaa bbb")
def test_getAll(self):
self.assertEqual(self.database.getAll(),[])
def test_getRandom(self):
self.database.add("aaa bbb")
self.database.add("aaa ccc")
value = self.database.getRandom()
self.assertTrue(value == "aaa bbb" or value == "aaa ccc")
def test_getRandom_exception(self):
from database import DataBaseException
with self.assertRaises(DataBaseException):
self.database.getRandom()
def test_add_no_multiplications(self):
self.database.add("aaa bbb")
self.database.add("aaa bbb")
self.database.add("aaa bbb")
self.database.add("aaa bbb")
self.assertEqual(len(self.database.getAll()),1)
