diff --git a/platform/platform-tests/testSrc/com/intellij/internal/ml/FeatureMappersTest.kt b/platform/platform-tests/testSrc/com/intellij/internal/ml/FeatureMappersTest.kt
index 4de0fce7b2d8..907b5892c0e3 100644
--- a/platform/platform-tests/testSrc/com/intellij/internal/ml/FeatureMappersTest.kt
+++ b/platform/platform-tests/testSrc/com/intellij/internal/ml/FeatureMappersTest.kt
@@ -119,4 +119,17 @@ class FeatureMappersTest {
     assertEquals(0.0, mapper.asArrayValue("private"))
     assertEquals(0.0, mapper.asArrayValue("null"))
   }
+
+  @Test
+  fun `categorical mapper should recognize enum elements`() {
+    val mapper = simpleCategoricalFeature.createMapper("public")
+    assertEquals(1.0, mapper.asArrayValue(Access.public))
+    assertEquals(0.0, mapper.asArrayValue(Access.private))
+    assertEquals(0.0, mapper.asArrayValue(null))
+  }
+
+  @Suppress("EnumEntryName")
+  private enum class Access {
+    private, public
+  }
 }
\ No newline at end of file