fix data profiler testcase

VIDA-NYU · Aug 26, 2024 · d7e92b9 · d7e92b9
1 parent f1af9ff
commit d7e92b9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_data_profiler.py b/tests/test_data_profiler.py
@@ -18,6 +18,9 @@ def test_profile_data():
                          'numeric_columns': [(0, 'show_id'), (7, 'release_year')],
                          'categorical_columns': [(1, 'type'), (2, 'title'), (3, 'director'), (4, 'cast'),
                                                  (5, 'country'), (6, 'date_added'), (8, 'duration'), (9, 'listed_in'),
-                                                 (10, 'description')]}
+                                                 (10, 'description')],
+                         'column_names': ['show_id', 'type', 'title', 'director', 'cast', 'country', 'date_added',
+                                          'release_year', 'duration', 'listed_in', 'description'],
+                        }
 
     assert actual_metadata == expected_metadata