Raise Semantic Error for bitwise op on complex types

dylon · Jun 21, 2022 · 2428888 · 2428888
1 parent 326383c
commit 2428888
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/lpython/semantics/python_ast_to_asr.cpp b/src/lpython/semantics/python_ast_to_asr.cpp
@@ -1307,6 +1307,11 @@ class CommonVisitor : public AST::BaseVisitor<Derived> {
 
         } else if (ASRUtils::is_complex(*dest_type)) {
 
+            if (op == ASR::binopType::BitAnd || op == ASR::binopType::BitOr || op == ASR::binopType::BitXor ||
+                op == ASR::binopType::BitLShift || op == ASR::binopType::BitRShift || op == ASR::binopType::Mod) {
+                throw SemanticError("Unsupported binary operation on complex: '" + ASRUtils::binop_to_str(op) + "'", loc);
+            }
+
             if (ASRUtils::expr_value(left) != nullptr && ASRUtils::expr_value(right) != nullptr) {
                 ASR::ComplexConstant_t *left0 = ASR::down_cast<ASR::ComplexConstant_t>(
                                                                 ASRUtils::expr_value(left));