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()
