test predict_connectivity: make sure to grab downstream neuron as target

navis-org · Feb 6, 2019 · ead35f2 · ead35f2
1 parent b050e56
commit ead35f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pymaid/tests/test_pymaid.py b/pymaid/tests/test_pymaid.py
@@ -748,7 +748,8 @@ def setUp(self):
         self.cn_table = pymaid.get_partners(config_test.test_skids[0],
                                             remote_instance=self.rm)
 
-        self.nB = pymaid.get_neuron(self.cn_table.iloc[0].skeleton_id,
+        # Must be downstream for predict_connectivity
+        self.nB = pymaid.get_neuron(self.cn_table[self.cn_table.relation=='downstream'].iloc[0].skeleton_id,
                                     remote_instance=self.rm)
 
         self.adj = pymaid.adjacency_matrix(