Update unit test

Hjorthmedh · Oct 23, 2023 · 65c7c6c · 65c7c6c
1 parent bab251a
commit 65c7c6c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/snudda/detect/detect.py b/snudda/detect/detect.py
@@ -2304,11 +2304,14 @@ def load_neuron(self, neuron_info, use_cache=True):
         if use_cache and neuron_id in self.neuron_cache:
             return self.neuron_cache[neuron_id]
 
-        if os.path.isfile(snudda_parse_path(neuron_info["morphology"], self.snudda_data)):
-            morphology_path = neuron_info["morphology"]
+        morph_path = snudda_parse_path(neuron_info["morphology"], self.snudda_data)
+        if os.path.isfile(morph_path):
+            morphology_path = morph_path
         else:
             morphology_path = None  # Get morpholog automatically from morphology_key
 
+        print(f"morphology_path = {morphology_path}")
+
         # Clone prototype neuron (it is centred, and not rotated)
         neuron = self.prototype_neurons[neuron_info["name"]].clone(parameter_key=neuron_info["parameterKey"],
                                                                    morphology_key=neuron_info["morphologyKey"],

diff --git a/snudda/plotting/plot_network.py b/snudda/plotting/plot_network.py
@@ -184,8 +184,9 @@ def load_neuron(self, neuron_info=None, neuron_id=None):
                                                                   load_morphology=True,
                                                                   virtual_neuron=False)
 
-        if os.path.isfile(snudda_parse_path(neuron_info["morphology"], self.snudda_data)):
-            morphology_path = neuron_info["morphology"]
+        morph_path = snudda_parse_path(neuron_info["morphology"], self.snudda_data)
+        if os.path.isfile(morph_path):
+            morphology_path = morph_path
         else:
             morphology_path = None  # Get morpholog automatically from morphology_key