def _traning_database_builder(self): from database import TrainingDataBaseBuilder builder = TrainingDataBaseBuilder() builder.addTxtFile("traning_sample.txt") return builder
class TrainingDataBaseBuilderTest(unittest.TestCase): def setUp(self): from database import TrainingDataBaseBuilder self.builder = TrainingDataBaseBuilder() def test_generateDataBase_empty(self): from database import TrainingDataBase,DataBaseException database = self.builder.generateDataBase() self.assertIsInstance(database,TrainingDataBase) self.assertEqual(database.getAll(),[]) def test_generateDataBase(self): from database import TrainingDataBase,DataBaseException from database import WordRecord self.builder.addTxtFile("traning_sample.txt") database = self.builder.generateDataBase() self.assertIsInstance(database,TrainingDataBase) data = database.getAll() self.assertIn(([WordRecord("abc"),WordRecord("abc"),WordRecord("abc")],[WordRecord("abc"),WordRecord("abc"),WordRecord("abc")]),data) self.assertIn(([WordRecord("abc"),WordRecord("abc"),WordRecord("ccc")],[WordRecord("abc"),WordRecord("abc"),WordRecord("kkk")]),data) self.assertIn(([WordRecord("zzz")],[WordRecord("bbb")]),data) self.assertNotIn(([WordRecord("aaa")],[WordRecord("aaa")]),data)
def setUp(self): from database import TrainingDataBaseBuilder self.builder = TrainingDataBaseBuilder()
if __name__ == "__main__": from jarvis import Jarvis from database import WordDataBaseBuilder from database import TrainingDataBaseBuilder from database import AbilitiesDataBaseBuilder from database import InnerVoiceDataBaseBuilder from interfaces import Console wd_builder = WordDataBaseBuilder() wd_builder.addTxtFile("learning_material/traning.txt") td_builder = TrainingDataBaseBuilder() td_builder.addTxtFile("learning_material/traning.txt") iv_builder = InnerVoiceDataBaseBuilder() iv_builder.addTxtFile("learning_material/inner_voices.txt") ab_builder = AbilitiesDataBaseBuilder() he = Jarvis() he.createWordsDataBase(wd_builder) he.createTrainingDataBase(td_builder) he.createAbilitiesDataBase(ab_builder) he.createInnerVoiceDatabase(iv_builder) he.train() console = Console() console.jarvis(he) console.start() he.start()