fix: Python generation for type casts

Safe-DS · Apr 6, 2024 · aa97a0b · aa97a0b
1 parent 8d22e67
commit aa97a0b
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/packages/safe-ds-lang/src/language/generation/safe-ds-python-generator.ts b/packages/safe-ds-lang/src/language/generation/safe-ds-python-generator.ts
@@ -47,6 +47,7 @@ import {
     isSdsTemplateStringInner,
     isSdsTemplateStringPart,
     isSdsTemplateStringStart,
+    isSdsTypeCast,
     isSdsWildcard,
     isSdsYield,
     SdsArgument,
@@ -907,6 +908,8 @@ export class SafeDsPythonGenerator {
             const referenceImport = this.createImportDataForReference(expression);
             frame.addImport(referenceImport);
             return traceToNode(expression)(referenceImport?.alias ?? this.getPythonNameOrDefault(declaration));
+        } else if (isSdsTypeCast(expression)) {
+            return traceToNode(expression)(this.generateExpression(expression.expression, frame));
         }
         /* c8 ignore next 2 */
         throw new Error(`Unknown expression type: ${expression.$type}`);

diff --git a/...thon/expressions/type casts/generated/tests/generation/expressions/typeCasts/gen_input.py b/...thon/expressions/type casts/generated/tests/generation/expressions/typeCasts/gen_input.py
@@ -0,0 +1,4 @@
+# Segments ---------------------------------------------------------------------
+
+def mySegment(p):
+    a = p
diff --git a/.../expressions/type casts/generated/tests/generation/expressions/typeCasts/gen_input.py.map b/.../expressions/type casts/generated/tests/generation/expressions/typeCasts/gen_input.py.map
diff --git a/packages/safe-ds-lang/tests/resources/generation/python/expressions/type casts/input.sdstest b/packages/safe-ds-lang/tests/resources/generation/python/expressions/type casts/input.sdstest
@@ -0,0 +1,5 @@
+package tests.generation.expressions.typeCasts
+
+segment mySegment(p: Int) {
+    val a = p as Int;
+}