Update tests/test_generators.py

Co-authored-by: Taylor Turner <[email protected]>
capitalone · Sep 27, 2023 · 8d65506 · 8d65506
1 parent c1bfa3d
commit 8d65506
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_generators.py b/tests/test_generators.py
@@ -405,15 +405,17 @@ def test_get_ordered_column_integration(self, mock_report, mock_warning):
             [4, "wqfed", "yellow", 7.775666, "2026-02-04"],
             [4, "wsde", "yellow", 7.818521, "2027-06-13"],
         ]
-        categories = [
+        expected_column_names = [
             "test_column_1",
             "test_column_2",
             "test_column_3",
             "test_column_4",
             "test_column_5",
         ]
 
-        expected_data = [dict(zip(categories, item)) for item in expected_array]
+        expected_data = [
+            dict(zip(expected_column_names, item)) for item in expected_array
+        ]
         expected_df = pd.DataFrame(expected_data)
 
         actual_df = generator.synthesize(20)