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)