test_neuron_model.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. import os
  2. import unittest
  3. from kalliope.core.Models import Neuron
  4. class TestNeuronModule(unittest.TestCase):
  5. def test_password_parameter(self):
  6. neuron_name = "test"
  7. neuron_parameters = {
  8. "password": "my secret",
  9. "parameter": "test"
  10. }
  11. neuron = Neuron()
  12. neuron.name = neuron_name
  13. neuron.parameters = neuron_parameters
  14. expected_result = "Neuron: name: test, parameters: {'password': '*****', 'parameter': 'test'}"
  15. self.assertEqual(neuron.__str__(), expected_result)
  16. def test_password_in_parameter(self):
  17. neuron_name = "test"
  18. neuron_parameters = {
  19. "password_parameter": "my secret",
  20. "parameter": "test"
  21. }
  22. neuron = Neuron()
  23. neuron.name = neuron_name
  24. neuron.parameters = neuron_parameters
  25. expected_result = "Neuron: name: test, parameters: {'parameter': 'test', 'password_parameter': '*****'}"
  26. self.assertEqual(neuron.__str__(), expected_result)
  27. if __name__ == '__main__':
  28. unittest.main()