adjust test_data_index.py

PlasmaControl · Nov 14, 2023 · e06c9e2 · e06c9e2
1 parent 8b8559a
commit e06c9e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/desc/compute/data_index.py b/desc/compute/data_index.py
@@ -129,6 +129,7 @@ def _decorator(func):
             "dim": dim,
             "coordinates": coordinates,
             "dependencies": deps,
+            "aliases": aliases,
         }
         for p in parameterization:
             flag = False

diff --git a/tests/test_data_index.py b/tests/test_data_index.py
@@ -92,8 +92,14 @@ def test_data_index_deps(self):
                                 if p in superclasses or p == base_class:
                                     queried_deps[base_class][name] = deps
 
+                                    for alias in data_index[base_class][name][
+                                        "aliases"
+                                    ]:
+                                        queried_deps[base_class][alias] = deps
+
         for p in data_index:
             for name, val in data_index[p].items():
+                print(name)
                 err_msg = f"Parameterization: {p}. Name: {name}."
                 deps = val["dependencies"]
                 data = set(deps["data"])