test_models.py 18 KB

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