Skip to content
Snippets Groups Projects
Commit 247f908a authored by Kenny Sharma's avatar Kenny Sharma
Browse files

[NRRPLT-4884] Load brain file for frontend.

The CLE no longer loads the brain module in terms of PyNN, but it
still needs to load the brain file source for the frontend to display.

Change-Id: I3cd7833387f6e542a05a29e651b38cf2031b6c54
parent e42c1245
No related branches found
No related tags found
No related merge requests found
......@@ -28,6 +28,12 @@ class MUSICPyNNControlAdapter(PyNNControlAdapter):
"""
self.__load_music_brain_proxies(**populations)
# load the brain source for the frontend to display, copied from parent class
logger.info("Saving brain source")
import hbp_nrp_cle.tf_framework.config as tf_config
with open(network_file) as source:
tf_config.brain_source = source.read()
@staticmethod
def __load_music_brain_proxies(**populations):
"""
......
import unittest
from hbp_nrp_music_interface.cle.MUSICPyNNControlAdapter import MUSICPyNNControlAdapter
import hbp_nrp_cle.tf_framework.config as tf_config
from mock import Mock, patch
from mock import Mock, patch, mock_open
import os
......@@ -22,9 +23,17 @@ class TestControlAdapter(unittest.TestCase):
@patch('os.environ.get', return_value='')
@patch('hbp_nrp_music_interface.cle.MUSICBrainLoader.load_proxies_from_xml')
def test_valid_xml(self, mocked_setup, mocked_environ, mocked_load):
mc = MUSICPyNNControlAdapter()
mc.load_brain('foo.xml')
m = mock_open()
m.return_value.read.return_value = 'mock brain source'
with patch('__builtin__.open', m, create=True):
mc = MUSICPyNNControlAdapter()
mc.load_brain('foo.xml')
mocked_environ.assert_called_once_with('NRP_MUSIC_DIRECTORY')
m.assert_called_once_with('foo.xml')
self.assertEquals(tf_config.brain_source, 'mock brain source')
if __name__ == "__main__":
unittest.main()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment