Skip to content

Commit 739eac3

Browse files
committed
Enable caching for performance tests by setting query limit to -1
1 parent fba780d commit 739eac3

1 file changed

Lines changed: 15 additions & 15 deletions

File tree

src/test/test_query_performance.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def test_02_neuron_part_queries(self):
128128
get_neurons_with_part_in,
129129
self.test_terms['antennal_lobe'],
130130
return_dataframe=False,
131-
limit=10
131+
limit=-1 # Changed to -1 to enable caching
132132
)
133133
print(f"NeuronsPartHere: {duration:.4f}s {'✅' if success else '❌'}")
134134
self.assertLess(duration, self.THRESHOLD_VERY_SLOW, "NeuronsPartHere exceeded threshold")
@@ -146,7 +146,7 @@ def test_03_synaptic_queries(self):
146146
get_neurons_with_synapses_in,
147147
test_term,
148148
return_dataframe=False,
149-
limit=10
149+
limit=-1 # Enable caching for performance tests
150150
)
151151
print(f"NeuronsSynaptic: {duration:.4f}s {'✅' if success else '❌'}")
152152
self.assertLess(duration, self.THRESHOLD_VERY_SLOW, "NeuronsSynaptic exceeded threshold")
@@ -156,7 +156,7 @@ def test_03_synaptic_queries(self):
156156
get_neurons_with_presynaptic_terminals_in,
157157
test_term,
158158
return_dataframe=False,
159-
limit=10
159+
limit=-1 # Enable caching for performance tests
160160
)
161161
print(f"NeuronsPresynapticHere: {duration:.4f}s {'✅' if success else '❌'}")
162162
self.assertLess(duration, self.THRESHOLD_VERY_SLOW, "NeuronsPresynapticHere exceeded threshold")
@@ -166,7 +166,7 @@ def test_03_synaptic_queries(self):
166166
get_neurons_with_postsynaptic_terminals_in,
167167
test_term,
168168
return_dataframe=False,
169-
limit=10
169+
limit=-1 # Enable caching for performance tests
170170
)
171171
print(f"NeuronsPostsynapticHere: {duration:.4f}s {'✅' if success else '❌'}")
172172
self.assertLess(duration, self.THRESHOLD_VERY_SLOW, "NeuronsPostsynapticHere exceeded threshold")
@@ -177,7 +177,7 @@ def test_03_synaptic_queries(self):
177177
get_neuron_neuron_connectivity,
178178
self.test_terms['connected_neuron'],
179179
return_dataframe=False,
180-
limit=10
180+
limit=-1 # Enable caching for performance tests
181181
)
182182
print(f"NeuronNeuronConnectivity: {duration:.4f}s {'✅' if success else '❌'}")
183183
self.assertLess(duration, self.THRESHOLD_SLOW, "NeuronNeuronConnectivity exceeded threshold")
@@ -195,7 +195,7 @@ def test_04_anatomy_hierarchy_queries(self):
195195
get_components_of,
196196
test_term,
197197
return_dataframe=False,
198-
limit=10
198+
limit=-1 # Enable caching for performance tests
199199
)
200200
print(f"ComponentsOf: {duration:.4f}s {'✅' if success else '❌'}")
201201
self.assertLess(duration, self.THRESHOLD_SLOW, "ComponentsOf exceeded threshold")
@@ -205,7 +205,7 @@ def test_04_anatomy_hierarchy_queries(self):
205205
get_parts_of,
206206
test_term,
207207
return_dataframe=False,
208-
limit=10
208+
limit=-1 # Enable caching for performance tests
209209
)
210210
print(f"PartsOf: {duration:.4f}s {'✅' if success else '❌'}")
211211
self.assertLess(duration, self.THRESHOLD_VERY_SLOW, "PartsOf exceeded threshold")
@@ -215,7 +215,7 @@ def test_04_anatomy_hierarchy_queries(self):
215215
get_subclasses_of,
216216
test_term,
217217
return_dataframe=False,
218-
limit=10
218+
limit=-1 # Enable caching for performance tests
219219
)
220220
print(f"SubclassesOf: {duration:.4f}s {'✅' if success else '❌'}")
221221
self.assertLess(duration, self.THRESHOLD_SLOW, "SubclassesOf exceeded threshold")
@@ -232,7 +232,7 @@ def test_05_new_queries(self):
232232
get_neuron_classes_fasciculating_here,
233233
self.test_terms['broad_root'],
234234
return_dataframe=False,
235-
limit=10
235+
limit=-1 # Enable caching for performance tests
236236
)
237237
print(f"NeuronClassesFasciculatingHere: {duration:.4f}s {'✅' if success else '❌'}")
238238
self.assertLess(duration, self.THRESHOLD_SLOW, "NeuronClassesFasciculatingHere exceeded threshold")
@@ -243,7 +243,7 @@ def test_05_new_queries(self):
243243
get_tracts_nerves_innervating_here,
244244
self.test_terms['antennal_lobe'],
245245
return_dataframe=False,
246-
limit=10
246+
limit=-1 # Enable caching for performance tests
247247
)
248248
print(f"TractsNervesInnervatingHere: {duration:.4f}s {'✅' if success else '❌'}")
249249
self.assertLess(duration, self.THRESHOLD_SLOW, "TractsNervesInnervatingHere exceeded threshold")
@@ -254,7 +254,7 @@ def test_05_new_queries(self):
254254
get_lineage_clones_in,
255255
self.test_terms['antennal_lobe'],
256256
return_dataframe=False,
257-
limit=10
257+
limit=-1 # Enable caching for performance tests
258258
)
259259
print(f"LineageClonesIn: {duration:.4f}s {'✅' if success else '❌'}")
260260
self.assertLess(duration, self.THRESHOLD_SLOW, "LineageClonesIn exceeded threshold")
@@ -265,7 +265,7 @@ def test_05_new_queries(self):
265265
get_images_neurons,
266266
self.test_terms['antennal_lobe'],
267267
return_dataframe=False,
268-
limit=10
268+
limit=-1 # Enable caching for performance tests
269269
)
270270
print(f"ImagesNeurons: {duration:.4f}s {'✅' if success else '❌'}")
271271
self.assertLess(duration, self.THRESHOLD_SLOW, "ImagesNeurons exceeded threshold")
@@ -276,7 +276,7 @@ def test_05_new_queries(self):
276276
get_images_that_develop_from,
277277
"FBbt_00001419", # neuroblast MNB - has 336 neuron images
278278
return_dataframe=False,
279-
limit=10
279+
limit=-1 # Enable caching for performance tests
280280
)
281281
print(f"ImagesThatDevelopFrom: {duration:.4f}s {'✅' if success else '❌'}")
282282
self.assertLess(duration, self.THRESHOLD_SLOW, "ImagesThatDevelopFrom exceeded threshold")
@@ -287,7 +287,7 @@ def test_05_new_queries(self):
287287
get_expression_pattern_fragments,
288288
"FBtp0000001", # expression pattern example
289289
return_dataframe=False,
290-
limit=10
290+
limit=-1 # Enable caching for performance tests
291291
)
292292
print(f"epFrag: {duration:.4f}s {'✅' if success else '❌'}")
293293
self.assertLess(duration, self.THRESHOLD_SLOW, "epFrag exceeded threshold")
@@ -303,7 +303,7 @@ def test_06_instance_queries(self):
303303
get_instances,
304304
self.test_terms['medulla'],
305305
return_dataframe=False,
306-
limit=5
306+
limit=-1 # Enable caching for performance tests
307307
)
308308
print(f"ListAllAvailableImages: {duration:.4f}s {'✅' if success else '❌'}")
309309
self.assertLess(duration, self.THRESHOLD_SLOW, "ListAllAvailableImages exceeded threshold")

0 commit comments

Comments
 (0)