diff --git a/idea/tests/org/jetbrains/kotlin/idea/parameterInfo/InlayTypeHintsTest.kt b/idea/tests/org/jetbrains/kotlin/idea/parameterInfo/InlayTypeHintsTest.kt
index 00ccd5874ef..7c0dad41520 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/parameterInfo/InlayTypeHintsTest.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/parameterInfo/InlayTypeHintsTest.kt
@@ -27,6 +27,7 @@ class InlayTypeHintsTest : KotlinLightCodeInsightFixtureTestCase() {
private fun checkLocalVariable(text: String) = check(text.trimIndent(), HintType.LOCAL_VARIABLE_HINT)
private fun checkPropertyHint(text: String) = check(text.trimIndent(), HintType.PROPERTY_HINT)
private fun checkFunctionHint(text: String) = check(text, HintType.FUNCTION_HINT)
+ private fun checkParameterTypeHint(text: String) = check(text, HintType.PARAMETER_TYPE_HINT)
fun testLocalVariableType() {
checkLocalVariable("""fun foo() { val a = listOf("a") }""")
@@ -285,4 +286,17 @@ class InlayTypeHintsTest : KotlinLightCodeInsightFixtureTestCase() {
"""
)
}
+
+ fun testParameterType() {
+ checkParameterTypeHint(
+ """
+ fun T.wrap(lambda: (T) -> T) {}
+ fun foo() {
+ 12.wrap { elem ->
+ elem
+ }
+ }
+ """
+ )
+ }
}