tests for semantic cache added

tests/test_semantic_cache.py

@@ -0,0 +1,58 @@
+from ontolearn.semantic_caching import run_semantic_cache, run_non_semantic_cache
+
+
+class TestSemanticCache:
+    def setup_method(self):
+        self.path_kg = "KGs/Family/family.owl" # path to the family datasets
+        self.path_kge = None
+        self.symbolic_reasoner = "Pellet"
+        self.neural_reasoner = "EBR"
+        self.num_concepts = 800
+        self.cache_size = 0.8*self.num_concepts
+        self.eviction = "LRU"
+        self.cache_type = "cold"
+
+    def run_cache_tests(self, cache_semantic, cache_non_semantic):
+        assert cache_semantic["hit_ratio"] >= cache_non_semantic["hit_ratio"], f"Expected semantic caching to have higher hit ratio, but got {cache_semantic['hit_ratio']} vs {cache_non_semantic['hit_ratio']}"
+        assert cache_semantic["miss_ratio"] <= cache_non_semantic["miss_ratio"], f"Expected semantic caching to have lower miss ratio, but got {cache_semantic['miss_ratio']} vs {cache_non_semantic['miss_ratio']}"
+
+    def test_run_time_and_jaccard(self):
+
+        cache_neural,_ =  run_semantic_cache(self.path_kg, self.path_kge, self.cache_size, self.neural_reasoner, self.eviction, 0, self.cache_type, True)
+        cache_symbolic,_ = run_semantic_cache(self.path_kg, self.path_kge, self.cache_size, self.symbolic_reasoner, self.eviction, 0, self.cache_type, True)
+
+        assert float(cache_neural["avg_jaccard"]) >= float(cache_neural["avg_jaccard_reas"]), "Expected average Jaccard similarity to be at least as good as reasoner-based retrieval."
+        assert float(cache_symbolic["avg_jaccard"]) >= float(cache_symbolic["avg_jaccard_reas"]), "Expected average Jaccard similarity to be at least as good as reasoner-based retrieval."
+        assert float(cache_symbolic["RT_cache"]) <= float(cache_symbolic["RT"]), "Expected runtime with cache to be less or equal to direct retrieval time."
+        assert float(cache_neural["RT_cache"]) <= float(cache_neural["RT"]), "Expected runtime with cache to be less or equal to direct retrieval time."
+
+
+    def test_cache_methods(self):
+        for reasoner in [self.neural_reasoner, self.symbolic_reasoner]:
+            cache_semantic,_ = run_semantic_cache(self.path_kg, self.path_kge, self.cache_size, reasoner, self.eviction, 0, self.cache_type, True)
+            cache_non_semantic,_ = run_non_semantic_cache(self.path_kg, self.path_kge, self.cache_size, reasoner, True)
+            self.run_cache_tests(cache_semantic, cache_non_semantic)
+
+    def test_cache_size(self):
+        cache_large,_ = run_semantic_cache(self.path_kg, self.path_kge, self.cache_size, self.neural_reasoner, self.eviction, 0, self.cache_type, True)
+
+        for k in [0.1, 0.2]:
+            cache_small,_ = run_semantic_cache(self.path_kg, self.path_kge, k * self.num_concepts, self.neural_reasoner, self.eviction, 0, self.cache_type, True)
+
+            assert cache_small["RT_cache"] >= cache_large["RT_cache"], "Expected runtime to decrease with larger cache."
+            assert cache_small["hit_ratio"] <= cache_large["hit_ratio"], f"Expected hit ratio to increase with cache size, but got {cache_small['hit_ratio']} vs {cache_large['hit_ratio']}"
+            assert cache_small["miss_ratio"] >= cache_large["miss_ratio"], f"Expected miss ratio to decrease with cache size, but got {cache_small['miss_ratio']} vs {cache_large['miss_ratio']}"
+
+    def test_eviction_strategy(self):
+        eviction_strategies = ["LRU", "FIFO", "LIFO", "MRU", "RP"]
+        results = {strategy: float(run_semantic_cache(self.path_kg, self.path_kge, self.cache_size, self.neural_reasoner, strategy, 10, self.cache_type, True)[0]["hit_ratio"]) for strategy in eviction_strategies}
+
+        for strategy, hit_ratio in results.items():
+            assert isinstance(hit_ratio, float), f"Hit ratio for {strategy} should be a float, but got {type(hit_ratio)}"
+
+        best_strategy = max(results, key=results.get)
+        assert best_strategy == "LRU", f"Expected LRU to be the best, but got {best_strategy}"
+
+        assert results, "No results were generated, possibly due to a failure in the cache evaluation process."
+        for strategy, hit_ratio in results.items():
+            assert 0.0 <= hit_ratio <= 1.0, f"Hit ratio for {strategy} is out of bounds: {hit_ratio}"