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