Add caching for painted domains and warm-up queries in performance tests

Author: Robbie1977

src/test/test_query_performance.py

@@ -375,6 +375,12 @@ def test_09_neuron_input_queries(self):
         print("="*80)
 
         # NeuronInputsTo
+        # Warm up cache with full results
+        try:
+            get_individual_neuron_inputs(self.test_terms['connected_neuron'], return_dataframe=False, limit=-1)
+        except Exception:
+            pass  # Ignore warm-up failures
+        
         result, duration, success = self._time_query(
             "NeuronInputsTo",
             get_individual_neuron_inputs,
@@ -606,6 +612,12 @@ def test_13_dataset_template_queries(self):
         dataset_term = 'VFBc_00101384'   # Example dataset
 
         # PaintedDomains - Template painted anatomy domains
+        # Warm up cache with full results
+        try:
+            get_painted_domains(template_term, return_dataframe=False, limit=-1)
+        except Exception:
+            pass  # Ignore warm-up failures
+        
         result, duration, success = self._time_query(
             "PaintedDomains",
             get_painted_domains,

src/vfbquery/cached_functions.py

@@ -309,6 +309,7 @@ def get_related_anatomy_cached(template_short_form: str, limit: int = -1, return
     """
     return _original_get_related_anatomy(template_short_form=template_short_form, limit=limit, return_dataframe=return_dataframe)
 
+@with_solr_cache('painted_domains')
 def get_painted_domains_cached(template_short_form: str, return_dataframe=True, limit: int = -1, force_refresh: bool = False):
     """
     Enhanced get_painted_domains with SOLR caching.

src/vfbquery/solr_result_cache.py

@@ -627,7 +627,8 @@ def wrapper(*args, **kwargs):
                                    'neurons_part_here', 'neurons_synaptic', 
                                    'neurons_presynaptic', 'neurons_postsynaptic',
                                    'expression_overlaps_here', 'anatomy_scrnaseq', 'aligned_datasets', 'terms_for_pub',
-                                   'individual_neuron_inputs', 'cluster_expression', 'expression_cluster', 'scrnaseq_dataset_data']
+                                   'individual_neuron_inputs', 'cluster_expression', 'expression_cluster', 'scrnaseq_dataset_data',
+                                   'painted_domains']
 
             # For neuron_neuron_connectivity_query, only cache when all parameters are defaults
             if query_type == 'neuron_neuron_connectivity_query':
@@ -665,7 +666,7 @@ def wrapper(*args, **kwargs):
                                    'neuron_region_connectivity_query', 'instances', 'templates', 'images_neurons',
                                    'images_that_develop_from', 'expression_pattern_fragments', 'expression_overlaps_here',
                                    'anatomy_scrnaseq', 'aligned_datasets', 'terms_for_pub', 'individual_neuron_inputs',
-                                   'cluster_expression', 'expression_cluster', 'scrnaseq_dataset_data']
+                                   'cluster_expression', 'expression_cluster', 'scrnaseq_dataset_data', 'painted_domains']
             if query_type in dataframe_query_types:
                 return_dataframe = kwargs.get('return_dataframe', True)  # Default is True
                 cache_term_id = f"{cache_term_id}_dataframe_{return_dataframe}"