test_neuron_model.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  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. print neuron.__str__()
  15. expected_result = "Neuron: name: test, parameters: {'password': '*****', 'parameter': 'test'}"
  16. self.assertEqual(neuron.__str__(), expected_result)
  17. def test_password_in_parameter(self):
  18. neuron_name = "test"
  19. neuron_parameters = {
  20. "password_parameter": "my secret",
  21. "parameter": "test"
  22. }
  23. neuron = Neuron()
  24. neuron.name = neuron_name
  25. neuron.parameters = neuron_parameters
  26. print neuron.__str__()
  27. expected_result = "Neuron: name: test, parameters: {'parameter': 'test', 'password_parameter': '*****'}"
  28. self.assertEqual(neuron.__str__(), expected_result)
  29. if __name__ == '__main__':
  30. unittest.main()