feat(python): add __init__ and myfunc to PythonContextTest class and …

…related imports.
unit-mesh · May 8, 2024 · 9f19dfb · 9f19dfb
1 parent 358f045
commit 9f19dfb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pycharm/src/main/kotlin/cc/unitmesh/python/context/PythonMethodContextBuilder.kt b/pycharm/src/main/kotlin/cc/unitmesh/python/context/PythonMethodContextBuilder.kt
@@ -11,7 +11,7 @@ import org.jetbrains.annotations.NotNull
 
 class PythonMethodContextBuilder : MethodContextBuilder {
     override fun getMethodContext(
-        @NotNull psiElement: PsiElement,
+        psiElement: PsiElement,
         includeClassContext: Boolean,
         gatherUsages: Boolean
     ): MethodContext? {
@@ -24,7 +24,7 @@ class PythonMethodContextBuilder : MethodContextBuilder {
         val returnType = returnStatementType?.name ?: ""
         val language = psiElement.language.displayName
         val enclosingClass = psiElement.containingClass
-        val signature = psiElement.qualifiedName
+        val signature = psiElement.name
         val name = psiElement.name
         val text = psiElement.text
         val paramNames = psiElement.parameterList.parameters.mapNotNull { it.name }

diff --git a/pycharm/src/test/kotlin/cc/unitmesh/python/context/PythonContextTest.kt b/pycharm/src/test/kotlin/cc/unitmesh/python/context/PythonContextTest.kt
@@ -30,7 +30,8 @@ print(p1.age) """
         assertEquals(classContext.format(), """'package: /foo.bar
 class Person {
   
-  
-}""");
+  + __init__
+  + myfunc
+}""")
     }
 }