def _word_database_builder(self): from database import WordDataBaseBuilder db_builder = WordDataBaseBuilder() db_builder.addTxtFile("sample.txt") return db_builder
class WordDataBaseBuilderTest(unittest.TestCase): def setUp(self): from database import WordDataBaseBuilder self.builder = WordDataBaseBuilder() def test_generateDataBase_empty(self): from database import WordDataBase,DataBaseException database = self.builder.generateDataBase() self.assertIsInstance(database,WordDataBase) with self.assertRaises(DataBaseException): database.wordId("aaa") def test_generateDataBase(self): from database import WordDataBase,DataBaseException from database import WordRecord self.builder.addTxtFile("sample.txt") database = self.builder.generateDataBase() self.assertIsInstance(database,WordDataBase) database.wordId(WordRecord("aaa")) database.wordId(WordRecord("bbb")) with self.assertRaises(DataBaseException): database.wordId(WordRecord("www"))
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()
def setUp(self): from database import WordDataBaseBuilder self.builder = WordDataBaseBuilder()