test_models.py 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484
  1. import unittest
  2. import mock
  3. from kalliope.core.Models.Tts import Tts
  4. from kalliope.core.Models.Trigger import Trigger
  5. from kalliope.core.Models.Stt import Stt
  6. from kalliope.core.Models.RestAPI import RestAPI
  7. from kalliope.core.Models.Dna import Dna
  8. from kalliope.core import LIFOBuffer
  9. from kalliope.core.Models.Settings import Settings
  10. from kalliope.core.Models import Neuron, Order, Synapse, Brain, Event, Resources
  11. from kalliope.core.Models.APIResponse import APIResponse
  12. from kalliope.core.Models.MatchedSynapse import MatchedSynapse
  13. class TestModels(unittest.TestCase):
  14. def setUp(self):
  15. # Init
  16. neuron1 = Neuron(name='neurone1', parameters={'var1': 'val1'})
  17. neuron2 = Neuron(name='neurone2', parameters={'var2': 'val2'})
  18. neuron3 = Neuron(name='neurone3', parameters={'var3': 'val3'})
  19. neuron4 = Neuron(name='neurone4', parameters={'var4': 'val4'})
  20. signal1 = Order(sentence="this is the sentence")
  21. signal2 = Order(sentence="this is the second sentence")
  22. signal3 = Order(sentence="that is part of the third sentence")
  23. self.synapse1 = Synapse(name="Synapse1", neurons=[neuron1, neuron2], signals=[signal1])
  24. self.synapse2 = Synapse(name="Synapse2", neurons=[neuron3, neuron4], signals=[signal2])
  25. self.synapse3 = Synapse(name="Synapse3", neurons=[neuron2, neuron4], signals=[signal3])
  26. self.all_synapse_list1 = [self.synapse1,
  27. self.synapse2,
  28. self.synapse3]
  29. self.all_synapse_list2 = [self.synapse2,
  30. self.synapse3]
  31. self.brain_test1 = Brain(synapses=self.all_synapse_list1)
  32. self.brain_test2 = Brain(synapses=self.all_synapse_list2)
  33. # this brain is the same as the first one
  34. self.brain_test3 = Brain(synapses=self.all_synapse_list1)
  35. self.settings_test = Settings(default_synapse="Synapse3")
  36. # clean the LiFO
  37. LIFOBuffer.lifo_list = list()
  38. def test_APIResponse(self):
  39. user_order = "user order"
  40. self.matched_synapse = MatchedSynapse(matched_synapse=self.synapse1, matched_order=user_order)
  41. api_response = APIResponse()
  42. api_response.user_order = user_order
  43. api_response.list_processed_matched_synapse = [self.matched_synapse]
  44. expected_result_serialize = {
  45. 'status': None,
  46. 'matched_synapses':
  47. [
  48. {
  49. 'matched_order': 'user order',
  50. 'neuron_module_list': [],
  51. 'synapse_name': 'Synapse1'
  52. }
  53. ],
  54. 'user_order': 'user order'
  55. }
  56. self.assertEqual(expected_result_serialize, api_response.serialize())
  57. expected_result_str = "{'status': None, 'matched_synapses': [{'matched_order': 'user order', " \
  58. "'neuron_module_list': [], 'synapse_name': 'Synapse1'}], 'user_order': 'user order'}"
  59. self.assertEqual(expected_result_str, api_response.__str__())
  60. def test_Brain(self):
  61. # test get synapse by name
  62. expect_result = self.synapse1
  63. synapse_name = "Synapse1"
  64. self.assertEqual(self.brain_test1.get_synapse_by_name(synapse_name), expect_result)
  65. # test equals
  66. self.assertTrue(self.brain_test1.__eq__(self.brain_test3))
  67. # test not equals
  68. self.assertFalse(self.brain_test1.__eq__(self.brain_test2))
  69. def test_Dna(self):
  70. # create DNA object
  71. dna1 = Dna(name="dna1", module_type="neuron", author="kalliope",
  72. kalliope_supported_version="0.4.4", tags="test")
  73. dna2 = Dna(name="dna2", module_type="neuron", author="community",
  74. kalliope_supported_version="0.4.2", tags="other")
  75. # this dna is exactly the same as the first one
  76. dna3 = Dna(name="dna1", module_type="neuron", author="kalliope",
  77. kalliope_supported_version="0.4.4", tags="test")
  78. expected_result_serialize = {
  79. 'kalliope_supported_version': '0.4.4',
  80. 'tags': 'test',
  81. 'type': 'neuron',
  82. 'name': 'dna1',
  83. 'author': 'kalliope'
  84. }
  85. expected_result_str = "{'kalliope_supported_version': '0.4.4', 'tags': 'test', 'type':" \
  86. " 'neuron', 'name': 'dna1', 'author': 'kalliope'}"
  87. self.assertEqual(expected_result_serialize, dna1.serialize())
  88. self.assertEqual(expected_result_str, dna1.__str__())
  89. self.assertTrue(dna1.__eq__(dna3))
  90. self.assertFalse(dna1.__eq__(dna2))
  91. def test_Event(self):
  92. event1 = Event(year=2017, month=12, day=31, week=53, day_of_week=2,
  93. hour=8, minute=30, second=0)
  94. event2 = Event(year=2018, month=11, day=30, week=25, day_of_week=4,
  95. hour=9, minute=40, second=0)
  96. # same as the event1
  97. event3 = Event(year=2017, month=12, day=31, week=53, day_of_week=2,
  98. hour=8, minute=30, second=0)
  99. expected_result_serialize = {
  100. 'event': {
  101. 'week': 53,
  102. 'second': 0,
  103. 'minute': 30,
  104. 'hour': 8,
  105. 'year': 2017,
  106. 'day': 31,
  107. 'day_of_week': 2,
  108. 'month': 12
  109. }
  110. }
  111. expected_result_str = "{'event': {'week': 53, 'second': 0, 'minute': 30, 'hour': 8, " \
  112. "'year': 2017, 'day': 31, 'day_of_week': 2, 'month': 12}}"
  113. self.assertEqual(expected_result_serialize, event1.serialize())
  114. self.assertEqual(expected_result_str, event1.__str__())
  115. self.assertTrue(event1.__eq__(event3))
  116. self.assertFalse(event1.__eq__(event2))
  117. def test_MatchedSynapse(self):
  118. user_order = "user order"
  119. matched_synapse1 = MatchedSynapse(matched_synapse=self.synapse1, matched_order=user_order)
  120. matched_synapse2 = MatchedSynapse(matched_synapse=self.synapse2, matched_order=user_order)
  121. matched_synapse3 = MatchedSynapse(matched_synapse=self.synapse1, matched_order=user_order)
  122. expected_result_serialize = {
  123. 'matched_order': 'user order',
  124. 'neuron_module_list': [],
  125. 'synapse_name': 'Synapse1'
  126. }
  127. expected_result_str = "{'matched_order': 'user order', 'neuron_module_list': [], 'synapse_name': 'Synapse1'}"
  128. self.assertEqual(expected_result_serialize, matched_synapse1.serialize())
  129. self.assertEqual(expected_result_str, matched_synapse1.__str__())
  130. self.assertTrue(matched_synapse1.__eq__(matched_synapse3))
  131. self.assertFalse(matched_synapse1.__eq__(matched_synapse2))
  132. # test neuron parameter loader is called
  133. with mock.patch("kalliope.core.NeuronParameterLoader.get_parameters") as mock_get_parameters:
  134. MatchedSynapse(matched_synapse=self.synapse1, matched_order=user_order, user_order=user_order)
  135. mock_get_parameters.assert_called_once_with(synapse_order=user_order,
  136. user_order=user_order)
  137. mock_get_parameters.reset_mock()
  138. def test_Neuron(self):
  139. neuron1 = Neuron(name="test", parameters={"key1": "val1", "key2": "val2"})
  140. neuron2 = Neuron(name="test", parameters={"key3": "val3", "key4": "val4"})
  141. neuron3 = Neuron(name="test", parameters={"key1": "val1", "key2": "val2"})
  142. expected_result_serialize = {'name': 'test', 'parameters': {'key2': 'val2', 'key1': 'val1'}}
  143. expected_result_str = "{'name': 'test', 'parameters': {'key2': 'val2', 'key1': 'val1'}}"
  144. self.assertEqual(expected_result_serialize, neuron1.serialize())
  145. self.assertEqual(expected_result_str, neuron1.__str__())
  146. self.assertTrue(neuron1.__eq__(neuron3))
  147. self.assertFalse(neuron1.__eq__(neuron2))
  148. # test password
  149. neuron_name = "test"
  150. neuron_parameters = {
  151. "password": "my secret",
  152. "parameter": "test"
  153. }
  154. neuron = Neuron()
  155. neuron.name = neuron_name
  156. neuron.parameters = neuron_parameters
  157. expected_result = "{'name': 'test', 'parameters': {'password': '*****', 'parameter': 'test'}}"
  158. self.assertEqual(neuron.__str__(), expected_result)
  159. neuron_name = "test"
  160. neuron_parameters = {
  161. "password_parameter": "my secret",
  162. "parameter": "test"
  163. }
  164. neuron = Neuron()
  165. neuron.name = neuron_name
  166. neuron.parameters = neuron_parameters
  167. expected_result = "{'name': 'test', 'parameters': {'parameter': 'test', 'password_parameter': '*****'}}"
  168. self.assertEqual(neuron.__str__(), expected_result)
  169. def test_Order(self):
  170. order1 = Order(sentence="this is an order")
  171. order2 = Order(sentence="this is an other order")
  172. order3 = Order(sentence="this is an order")
  173. expected_result_serialize = {'order': 'this is an order'}
  174. expected_result_str = "{'order': 'this is an order'}"
  175. self.assertEqual(expected_result_serialize, order1.serialize())
  176. self.assertEqual(expected_result_str, order1.__str__())
  177. self.assertTrue(order1.__eq__(order3))
  178. self.assertFalse(order1.__eq__(order2))
  179. def test_Resources(self):
  180. resource1 = Resources(neuron_folder="/path/neuron", stt_folder="/path/stt",
  181. tts_folder="/path/tts", trigger_folder="/path/trigger")
  182. resource2 = Resources(neuron_folder="/other_path/neuron", stt_folder="/other_path/stt",
  183. tts_folder="/other_path/tts", trigger_folder="/other_path/trigger")
  184. resource3 = Resources(neuron_folder="/path/neuron", stt_folder="/path/stt",
  185. tts_folder="/path/tts", trigger_folder="/path/trigger")
  186. expected_result_serialize = {
  187. 'tts_folder': '/path/tts',
  188. 'neuron_folder': '/path/neuron',
  189. 'stt_folder': '/path/stt',
  190. 'trigger_folder': '/path/trigger'
  191. }
  192. expected_result_str = "{'tts_folder': '/path/tts', 'neuron_folder': '/path/neuron', " \
  193. "'stt_folder': '/path/stt', 'trigger_folder': '/path/trigger'}"
  194. self.assertEqual(expected_result_serialize, resource1.serialize())
  195. self.assertEqual(expected_result_str, resource1.__str__())
  196. self.assertTrue(resource1.__eq__(resource3))
  197. self.assertFalse(resource1.__eq__(resource2))
  198. def test_RestAPI(self):
  199. rest_api1 = RestAPI(password_protected=True, login="admin", password="password", active=True,
  200. port=5000, allowed_cors_origin="*")
  201. rest_api2 = RestAPI(password_protected=False, active=False,
  202. port=5000, allowed_cors_origin=None)
  203. rest_api3 = RestAPI(password_protected=True, login="admin", password="password", active=True,
  204. port=5000, allowed_cors_origin="*")
  205. expected_result_serialize = {
  206. 'password_protected': True,
  207. 'port': 5000,
  208. 'active': True,
  209. 'allowed_cors_origin': '*',
  210. 'password': 'password',
  211. 'login': 'admin'
  212. }
  213. expected_result_str = "{'password_protected': True, 'port': 5000, 'active': True, 'allowed_cors_origin': '*', " \
  214. "'password': 'password', 'login': 'admin'}"
  215. self.assertEqual(expected_result_serialize, rest_api1.serialize())
  216. self.assertEqual(expected_result_str, rest_api1.__str__())
  217. self.assertTrue(rest_api1.__eq__(rest_api3))
  218. self.assertFalse(rest_api1.__eq__(rest_api2))
  219. def test_Settings(self):
  220. with mock.patch('platform.machine', return_value='pumpkins'):
  221. rest_api1 = RestAPI(password_protected=True, login="admin", password="password", active=True,
  222. port=5000, allowed_cors_origin="*")
  223. setting1 = Settings(default_tts_name="pico2wav",
  224. default_stt_name="google",
  225. default_trigger_name="swoyboy",
  226. ttss=["ttts"],
  227. stts=["stts"],
  228. random_wake_up_answers=["yes"],
  229. random_wake_up_sounds=None,
  230. play_on_ready_notification=False,
  231. on_ready_answers=None,
  232. on_ready_sounds=None,
  233. triggers=["snowboy"],
  234. rest_api=rest_api1,
  235. cache_path="/tmp/kalliope",
  236. default_synapse="default_synapse",
  237. resources=None,
  238. variables={"key1": "val1"})
  239. setting2 = Settings(default_tts_name="accapela",
  240. default_stt_name="bing",
  241. default_trigger_name="swoyboy",
  242. ttss=["ttts"],
  243. stts=["stts"],
  244. random_wake_up_answers=["no"],
  245. random_wake_up_sounds=None,
  246. play_on_ready_notification=False,
  247. on_ready_answers=None,
  248. on_ready_sounds=None,
  249. triggers=["snowboy"],
  250. rest_api=rest_api1,
  251. cache_path="/tmp/kalliope_tmp",
  252. default_synapse="my_default_synapse",
  253. resources=None,
  254. variables={"key1": "val1"})
  255. setting3 = Settings(default_tts_name="pico2wav",
  256. default_stt_name="google",
  257. default_trigger_name="swoyboy",
  258. ttss=["ttts"],
  259. stts=["stts"],
  260. random_wake_up_answers=["yes"],
  261. random_wake_up_sounds=None,
  262. play_on_ready_notification=False,
  263. on_ready_answers=None,
  264. on_ready_sounds=None,
  265. triggers=["snowboy"],
  266. rest_api=rest_api1,
  267. cache_path="/tmp/kalliope",
  268. default_synapse="default_synapse",
  269. resources=None,
  270. variables={"key1": "val1"})
  271. expected_result_serialize = {
  272. 'default_synapse': 'default_synapse',
  273. 'default_tts_name': 'pico2wav',
  274. 'rest_api':
  275. {
  276. 'password_protected': True,
  277. 'port': 5000,
  278. 'active': True,
  279. 'allowed_cors_origin': '*',
  280. 'password': 'password',
  281. 'login': 'admin'
  282. },
  283. 'play_on_ready_notification': False,
  284. 'default_stt_name': 'google',
  285. 'kalliope_version': '0.4.4b',
  286. 'random_wake_up_sounds': None,
  287. 'on_ready_answers': None,
  288. 'default_trigger_name': 'swoyboy',
  289. 'cache_path': '/tmp/kalliope',
  290. 'stts': ['stts'],
  291. 'machine': 'pumpkins',
  292. 'random_wake_up_answers': ['yes'],
  293. 'on_ready_sounds': None,
  294. 'ttss': ['ttts'],
  295. 'variables': {'key1': 'val1'},
  296. 'resources': None,
  297. 'triggers': ['snowboy']
  298. }
  299. expected_result_str = "{'default_synapse': 'default_synapse', 'default_tts_name': 'pico2wav', 'rest_api': " \
  300. "{'password_protected': True, 'port': 5000, 'active': True, 'allowed_cors_origin': " \
  301. "'*', 'password': 'password', 'login': 'admin'}, 'play_on_ready_notification': " \
  302. "False, 'default_stt_name': 'google', 'kalliope_version': '0.4.4b', " \
  303. "'random_wake_up_sounds': None, 'on_ready_answers': None, 'default_trigger_name': " \
  304. "'swoyboy', 'cache_path': '/tmp/kalliope', 'stts': ['stts'], 'machine': 'pumpkins', " \
  305. "'random_wake_up_answers': ['yes'], 'on_ready_sounds': None, 'ttss': ['ttts'], " \
  306. "'variables': {'key1': 'val1'}, 'resources': None, 'triggers': ['snowboy']}"
  307. self.assertEqual(expected_result_serialize, setting1.serialize())
  308. self.assertEqual(expected_result_str, setting1.__str__())
  309. self.assertTrue(setting1.__eq__(setting3))
  310. self.assertFalse(setting1.__eq__(setting2))
  311. def test_Stt(self):
  312. stt1 = Stt(name="stt1", parameters={"key1": "val1"})
  313. stt2 = Stt(name="stt2", parameters={"key2": "val2"})
  314. stt3 = Stt(name="stt1", parameters={"key1": "val1"})
  315. expected_result_serialize = {'name': 'stt1', 'parameters': {'key1': 'val1'}}
  316. expected_result_str = "{'name': 'stt1', 'parameters': {'key1': 'val1'}}"
  317. self.assertEqual(expected_result_serialize, stt1.serialize())
  318. self.assertEqual(expected_result_str, stt1.__str__())
  319. self.assertTrue(stt1.__eq__(stt3))
  320. self.assertFalse(stt1.__eq__(stt2))
  321. def test_Synapse(self):
  322. neuron1 = Neuron(name='neurone1', parameters={'var1': 'val1'})
  323. neuron2 = Neuron(name='neurone2', parameters={'var2': 'val2'})
  324. neuron3 = Neuron(name='neurone3', parameters={'var3': 'val3'})
  325. neuron4 = Neuron(name='neurone4', parameters={'var4': 'val4'})
  326. signal1 = Order(sentence="this is the sentence")
  327. signal2 = Order(sentence="this is the second sentence")
  328. synapse1 = Synapse(name="Synapse1", neurons=[neuron1, neuron2], signals=[signal1])
  329. synapse2 = Synapse(name="Synapse2", neurons=[neuron3, neuron4], signals=[signal2])
  330. synapse3 = Synapse(name="Synapse1", neurons=[neuron1, neuron2], signals=[signal1])
  331. expected_result_serialize = {
  332. 'signals': [
  333. {
  334. 'order': 'this is the sentence'
  335. }
  336. ],
  337. 'neurons': [
  338. {
  339. 'name': 'neurone1',
  340. 'parameters': {
  341. 'var1': 'val1'
  342. }
  343. },
  344. {
  345. 'name': 'neurone2',
  346. 'parameters':
  347. {
  348. 'var2': 'val2'
  349. }
  350. }
  351. ],
  352. 'name': 'Synapse1'
  353. }
  354. expected_result_str = "{'signals': [{'order': 'this is the sentence'}], 'neurons': [{'name': 'neurone1', " \
  355. "'parameters': {'var1': 'val1'}}, {'name': 'neurone2', 'parameters': {'var2': " \
  356. "'val2'}}], 'name': 'Synapse1'}"
  357. self.assertEqual(expected_result_serialize, synapse1.serialize())
  358. self.assertEqual(expected_result_str, synapse1.__str__())
  359. self.assertTrue(synapse1.__eq__(synapse3))
  360. self.assertFalse(synapse1.__eq__(synapse2))
  361. def test_Trigger(self):
  362. trigger1 = Trigger(name="trigger1", parameters={"key1": "val1"})
  363. trigger2 = Trigger(name="trigger2", parameters={"key2": "val2"})
  364. trigger3 = Trigger(name="trigger1", parameters={"key1": "val1"})
  365. expected_result_serialize = {'name': 'trigger1', 'parameters': {'key1': 'val1'}}
  366. expected_result_str = "{'name': 'trigger1', 'parameters': {'key1': 'val1'}}"
  367. self.assertEqual(expected_result_serialize, trigger1.serialize())
  368. self.assertEqual(expected_result_str, trigger1.__str__())
  369. self.assertTrue(trigger1.__eq__(trigger3))
  370. self.assertFalse(trigger1.__eq__(trigger2))
  371. def test_Tts(self):
  372. tts1 = Tts(name="tts1", parameters={"key1": "val1"})
  373. tts2 = Tts(name="tts2", parameters={"key2": "val2"})
  374. tts3 = Tts(name="tts1", parameters={"key1": "val1"})
  375. expected_result_serialize = {'name': 'tts1', 'parameters': {'key1': 'val1'}}
  376. expected_result_str = "{'name': 'tts1', 'parameters': {'key1': 'val1'}}"
  377. self.assertEqual(expected_result_serialize, tts1.serialize())
  378. self.assertEqual(expected_result_str, tts1.__str__())
  379. self.assertTrue(tts1.__eq__(tts3))
  380. self.assertFalse(tts1.__eq__(tts2))