Преглед на файлове

add a test in neuron launcher for reserved keys

nico преди 7 години
родител
ревизия
329934bd79
променени са 2 файла, в които са добавени 24 реда и са изтрити 7 реда
  1. 23 6
      Tests/test_neuron_launcher.py
  2. 1 1
      kalliope/core/NeuronLauncher.py

+ 23 - 6
Tests/test_neuron_launcher.py

@@ -209,47 +209,64 @@ class TestNeuronLauncher(unittest.TestCase):
         self.assertEqual(expected_result, NeuronLauncher._replace_brackets_by_loaded_parameter(neuron_parameters,
                                                                                                loaded_parameters))
 
+        # the parameter is a reserved key. for example from_answer_link from the neurotransmitter
+        list_reserved_keys = ["say_template", "file_template", "kalliope_memory", "from_answer_link"]
+
+        for reserved_key in list_reserved_keys:
+            neuron_parameters = {
+                reserved_key: "this is a value with {{ 'brackets '}}"
+            }
+
+            loaded_parameters = dict()
+
+            expected_result = {
+                reserved_key: "this is a value with {{ 'brackets '}}"
+            }
+
+            self.assertEqual(expected_result, NeuronLauncher._replace_brackets_by_loaded_parameter(neuron_parameters,
+                                                                                               loaded_parameters))
+
     def test_parameters_are_available_in_loaded_parameters(self):
         # the parameter in bracket is available in the dict
         string_parameters = "this is a {{ parameter1 }}"
         loaded_parameters = {"parameter1": "value"}
 
         self.assertTrue(NeuronLauncher._neuron_parameters_are_available_in_loaded_parameters(string_parameters,
-                                                                                            loaded_parameters))
+                                                                                             loaded_parameters))
 
         # the parameter in bracket is NOT available in the dict
         string_parameters = "this is a {{ parameter1 }}"
         loaded_parameters = {"parameter2": "value"}
 
         self.assertFalse(NeuronLauncher._neuron_parameters_are_available_in_loaded_parameters(string_parameters,
-                                                                                             loaded_parameters))
+                                                                                              loaded_parameters))
 
         # the string_parameters doesn't contains bracket in bracket is available in the dict
         string_parameters = "this is a {{ parameter1 }}"
         loaded_parameters = {"parameter1": "value"}
 
         self.assertTrue(NeuronLauncher._neuron_parameters_are_available_in_loaded_parameters(string_parameters,
-                                                                                            loaded_parameters))
+                                                                                             loaded_parameters))
 
         # the string_parameters contains 2 parameters available in the dict
         string_parameters = "this is a {{ parameter1 }} and this is {{ parameter2 }}"
         loaded_parameters = {"parameter1": "value", "parameter2": "other value"}
 
         self.assertTrue(NeuronLauncher._neuron_parameters_are_available_in_loaded_parameters(string_parameters,
-                                                                                            loaded_parameters))
+                                                                                             loaded_parameters))
 
         # the string_parameters contains 2 parameters and one of them is not available in the dict
         string_parameters = "this is a {{ parameter1 }} and this is {{ parameter2 }}"
         loaded_parameters = {"parameter1": "value", "parameter3": "other value"}
 
         self.assertFalse(NeuronLauncher._neuron_parameters_are_available_in_loaded_parameters(string_parameters,
-                                                                                             loaded_parameters))
+                                                                                              loaded_parameters))
 
 
 if __name__ == '__main__':
     unittest.main()
 
     # suite = unittest.TestSuite()
-    # suite.addTest(TestNeuronLauncher("test_start_neuron"))
+    # suite.addTest(TestNeuronLauncher("test_replace_brackets_by_loaded_parameter"))
     # runner = unittest.TextTestRunner()
     # runner.run(suite)

+ 1 - 1
kalliope/core/NeuronLauncher.py

@@ -98,7 +98,7 @@ class NeuronLauncher:
         if isinstance(neuron_parameters, dict):
             returned_dict = dict()
             for key, value in neuron_parameters.items():
-                # following keys are reserved for the TTS
+                # following keys are reserved by kalliope core
                 if key in "say_template" or key in "file_template" or key in "kalliope_memory" \
                         or key in "from_answer_link":
                     returned_dict[key] = value