Minor changes

cpg-language-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/ExpressionHandler.kt

@@ -28,6 +28,7 @@ package de.fraunhofer.aisec.cpg.frontends.python
 import de.fraunhofer.aisec.cpg.graph.*
 import de.fraunhofer.aisec.cpg.graph.declarations.ImportDeclaration
 import de.fraunhofer.aisec.cpg.graph.declarations.MethodDeclaration
+import de.fraunhofer.aisec.cpg.graph.statements.expressions.AssignExpression
 import de.fraunhofer.aisec.cpg.graph.statements.expressions.Expression
 import de.fraunhofer.aisec.cpg.graph.statements.expressions.MemberExpression
 import de.fraunhofer.aisec.cpg.graph.statements.expressions.ProblemExpression
@@ -87,11 +88,12 @@ class ExpressionHandler(frontend: PythonLanguageFrontend) :
 
     /**
      * Translates a Python (Named
-     * Expression)[https://docs.python.org/3/library/ast.html#ast.NamedExpr] into a [Expression].
+     * Expression)[https://docs.python.org/3/library/ast.html#ast.NamedExpr] into an
+     * [AssignExpression].
      *
      * As opposed to the Assign node, both target and value must be single nodes.
      */
-    private fun handleNamedExpr(node: Python.AST.NamedExpr): Expression {
+    private fun handleNamedExpr(node: Python.AST.NamedExpr): AssignExpression {
         val assignExpression =
             newAssignExpression(
                 operatorCode = ":=",

cpg-language-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/StatementHandler.kt

@@ -30,6 +30,7 @@ import de.fraunhofer.aisec.cpg.frontends.python.PythonLanguage.Companion.MODIFIE
 import de.fraunhofer.aisec.cpg.graph.*
 import de.fraunhofer.aisec.cpg.graph.Annotation
 import de.fraunhofer.aisec.cpg.graph.declarations.*
+import de.fraunhofer.aisec.cpg.graph.statements.AssertStatement
 import de.fraunhofer.aisec.cpg.graph.statements.DeclarationStatement
 import de.fraunhofer.aisec.cpg.graph.statements.Statement
 import de.fraunhofer.aisec.cpg.graph.statements.expressions.Block
@@ -80,13 +81,14 @@ class StatementHandler(frontend: PythonLanguageFrontend) :
 
     /**
      * Translates a Python (https://docs.python.org/3/library/ast.html#ast.Assert] into a
-     * [Statement].
+     * [AssertStatement].
      */
-    private fun handleAssert(node: Python.AST.Assert): Statement {
+    private fun handleAssert(node: Python.AST.Assert): AssertStatement {
         val assertStatement = newAssertStatement(rawNode = node)
         val testExpression = frontend.expressionHandler.handle(node.test)
         assertStatement.condition = testExpression
         node.msg?.let { assertStatement.message = frontend.expressionHandler.handle(it) }
+
         return assertStatement
     }