Simplify cache module (#52)

Author: namikmesic

src/cache.py

@@ -9,24 +9,19 @@ def __init__(self):
 
     def is_cached(self, key: str):
         """Check if key is cached."""
-        if key in self._cache:
-            return True
-        return False
+        return key in self._cache
 
     def store_key_value(self, key: str, value):
         """Stores key-value in cache dict."""
         self._cache[key] = value
 
     def retrieve_key_value(self, key: str):
         """Retrieves key value from cache."""
-        if self.is_cached(key):
-            return self._cache.get(key)
-        return None
+        return self._cache.get(key)
 
     def remove_key_from_cache(self, key: str):
         """Deletes a key from cache."""
-        if self.is_cached(key):
-            del self._cache[key]
+        return self._cache.pop(key, None)
 
     def clear_cache(self):
         """Clears the entire cache"""

src/test_cache.py

@@ -42,6 +42,6 @@ def test_is_cached(self):
 
     def test_clear_cache(self):
         """Tests that the cache attribute is cleared when the clear_cache method is called"""
-        self.cache._cache["dummy"] = "data"
+        self.cache.store_key_value("key", "value")
         self.cache.clear_cache()
-        self.assertTrue(len(self.cache._cache) == 0)
+        self.assertEqual(self.cache._cache, {})