// Mimic [psiForUast] in FIR UAST
internal fun KaSession.psiForTest(symbol: KaSymbol, project: Project): PsiElement? {
    return when (symbol.origin) {
        KaSymbolOrigin.LIBRARY -> {
            findPsi(symbol, project) ?: symbol.psi
        }
        KaSymbolOrigin.SUBSTITUTION_OVERRIDE, KaSymbolOrigin.INTERSECTION_OVERRIDE -> {
            psiForTest((symbol as KaCallableSymbol).unwrapFakeOverrides, project)
        }