fix: test cases

Signed-off-by: s0nicboOm <i.kushalbatra@gmail.com>
numaproj · Sep 12, 2023 · 965c9c5 · 965c9c5
1 parent b03f9f9
commit 965c9c5
Show file tree

Hide file tree

Showing 4 changed files with 221 additions and 176 deletions.
diff --git a/numalogic/registry/redis_registry.py b/numalogic/registry/redis_registry.py
@@ -385,6 +385,8 @@ def save_multiple(
             metadata: additional metadata surrounding the artifact that needs to be saved.
         """
         dict_model_ver = {}
+        if len(list_artifacts) != len(list_dkeys):
+            raise IndexError("artifact list and dkeys list should have same length!")
         try:
             with self.client.pipeline(transaction=self.transactional) as pipe:
                 pipe.multi()
@@ -405,7 +407,5 @@ def save_multiple(
             _LOGGER.info("Successfully saved all the artifacts with: %s", dict_model_ver)
         except RedisError as err:
             raise RedisRegistryError(f"{err.__class__.__name__} raised") from err
-        except IndexError as index_err:
-            raise RedisRegistryError(f"{index_err.__class__.__name__} raised") from index_err
         else:
             return dict_model_ver