test: final test update

lucas-nelson-uiuc · Nov 28, 2024 · 6be5ac8 · 6be5ac8
1 parent b54ffd7
commit 6be5ac8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_selector.py b/tests/test_selector.py
@@ -72,7 +72,7 @@ def test_numeric(self, sample_data):
 
     def test_temporal(self, sample_data):
         selected_columns = sample_data.select(cs.temporal()).columns
-        expected_columns = ["seasons"]
+        expected_columns = ["birth_date", "original_air_date"]
         assert set(selected_columns).difference(expected_columns) == set()
 
     def test_date(self, sample_data):