From e06c9e2f278ad8e5f4ce29c5bc9668bb0a86befc Mon Sep 17 00:00:00 2001
From: Kian Orr <orrk931@gmail.com>
Date: Mon, 13 Nov 2023 21:19:21 -0500
Subject: [PATCH] adjust test_data_index.py

---
 desc/compute/data_index.py | 1 +
 tests/test_data_index.py   | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/desc/compute/data_index.py b/desc/compute/data_index.py
index e1a998c14e..2944aee15d 100644
--- 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
index 325cba7a76..7ce9f4a774 100644
--- 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"])