test_neuron_launcher.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import unittest
  2. import mock
  3. from kalliope.core.Models.Resources import Resources
  4. from kalliope.core.NeuronLauncher import NeuronLauncher
  5. from kalliope.core.ConfigurationManager import SettingLoader
  6. from kalliope.core.Models.Neuron import Neuron
  7. class TestNeuronLauncher(unittest.TestCase):
  8. """
  9. Class to test Launchers Classes (TriggerLauncher, SynapseLauncher, NeuronLauncher) and methods
  10. """
  11. def setUp(self):
  12. pass
  13. ####
  14. # Neurons Launcher
  15. def test_start_neuron(self):
  16. """
  17. Test the Neuron Launcher trying to start a Neuron
  18. """
  19. neuron = Neuron(name='neurone1', parameters={'var1': 'val1'})
  20. sl = SettingLoader()
  21. resources = Resources(neuron_folder='/var/tmp/test/resources')
  22. sl.settings.resources = resources
  23. with mock.patch("kalliope.core.Utils.get_dynamic_class_instantiation") as mock_get_class_instantiation:
  24. NeuronLauncher.start_neuron(neuron=neuron)
  25. mock_get_class_instantiation.assert_called_once_with(package_name="neurons",
  26. module_name=neuron.name,
  27. parameters=neuron.parameters,
  28. resources_dir=sl.settings.resources.neuron_folder)
  29. mock_get_class_instantiation.reset_mock()