Test that ehrapycat_day_icu_intime is in result

namsaraeva · Jun 14, 2023 · 143e69f · 143e69f
1 parent 01e1e38
commit 143e69f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/tools/test_features_ranking.py b/tests/tools/test_features_ranking.py
@@ -159,6 +159,9 @@ def test_evaluate_categorical_features(self):
             assert "service_unit" not in names
             assert "ehrapycat_service_unit" not in names
 
+            # Check that the only other categorical feature is in the results
+            assert "ehrapycat_day_icu_intime" in names
+
 
 class TestRankFeaturesGroups():
     def test_real_dataset(self):