From 2ac5b4d99ca26e3ad9fd62653f0d86a1fa64bb8d Mon Sep 17 00:00:00 2001
From: "Vitaliy.Bibaev" <vitaliy.bibaev@jetbrains.com>
Date: Tue, 3 Sep 2019 17:33:11 +0300
Subject: [PATCH] [stats-collector] Add test: feature mapper understands enum
 elements

GitOrigin-RevId: e96bd2f8fb25fc396e3f940a4e142ed85b99f345
---
 .../com/intellij/internal/ml/FeatureMappersTest.kt  | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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