Fixing unit test

Hjorthmedh · Jul 18, 2024 · fc7b47c · fc7b47c
1 parent 20a872f
commit fc7b47c
Show file tree

Hide file tree

Showing 3 changed files with 23,079 additions and 48 deletions.
diff --git a/examples/parallel/KTH_PDC/neuromodulation/dspn/Dardel_analyse_dspn.ipynb b/examples/parallel/KTH_PDC/neuromodulation/dspn/Dardel_analyse_dspn.ipynb
diff --git a/snudda/simulate/simulate.py b/snudda/simulate/simulate.py
@@ -1735,7 +1735,14 @@ def add_synapse_current_recording(self, source_id, dest_id):
 
         return syn_ctr
 
-    def add_rxd_concentration_recording(self, species: str, neuron_id: int, region, sec_type, sec_id, sec_x):
+    def add_rxd_concentration_recording(self, species: str, neuron_id: int, region, sec_id, sec_x):
+
+        if sec_id == -1:
+            sec_type = "soma"
+        elif sec_id >= 0:
+            sec_type = "dend"
+        else:
+            sec_type = "axon"
 
         if self.neurons[neuron_id].modulation is None:
             raise ValueError(f"No modulation specified for neuron {self.neurons[neuron_id].name} ({neuron_id})")

diff --git a/tests/test_neuron_modulation.py b/tests/test_neuron_modulation.py
@@ -48,40 +48,34 @@ def test_reaction(self):
 
         self.sim.add_rxd_concentration_recording(species="DA", neuron_id=0,
                                                  region="soma_internal",
-                                                 sec_type="soma",
-                                                 sec_id=0,
+                                                 sec_id=-1,
                                                  sec_x=0.5)
 
         self.sim.add_rxd_concentration_recording(species="B", neuron_id=0,
                                                  region="soma_internal",
-                                                 sec_type="soma",
-                                                 sec_id=0,
+                                                 sec_id=-1,
                                                  sec_x=0.5)
 
         self.sim.add_rxd_concentration_recording(species="PKA", neuron_id=0,
                                                  region="soma_internal",
-                                                 sec_type="soma",
-                                                 sec_id=0,
+                                                 sec_id=-1,
                                                  sec_x=0.5)
 
         self.sim.add_density_mechanism_recording(density_mechanism="kirrxd",
                                                  variable="modulation_factor",
                                                  neuron_id=0,
-                                                 sec_type="soma",
-                                                 sec_id=0,
+                                                 sec_id=-1,
                                                  sec_x=0.5)
 
         self.sim.add_density_mechanism_recording(density_mechanism="kirrxd",
                                                  variable="m",
                                                  neuron_id=0,
-                                                 sec_type="soma",
-                                                 sec_id=0,
+                                                 sec_id=-1,
                                                  sec_x=0.5)
 
         self.sim.add_membrane_recording(variable="PKAi",
                                         neuron_id=0,
-                                        sec_type="soma",
-                                        sec_id=0,
+                                        sec_id=-1,
                                         sec_x=0.5)
 
         # Add DA synapse