[stats-collector] Add test: feature mapper understands enum elements

GitOrigin-RevId: e96bd2f8fb25fc396e3f940a4e142ed85b99f345
This commit is contained in:
Vitaliy.Bibaev
2019-09-03 17:32:59 +00:00
committed by intellij-monorepo-bot
parent df8fb7140e
commit 2ac5b4d99c
@@ -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
}
}