Add Binops

Philipp15b · Sep 24, 2023 · c6eb165 · c6eb165
1 parent c65b762
commit c6eb165
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/probably/prism/translate.py b/probably/prism/translate.py
@@ -134,6 +134,10 @@ def expression_prism(expr: Expr) -> str:
             return f"({lhs}) <= ({rhs})"
         elif expr.operator == Binop.LT:
             return f"({lhs}) < ({rhs})"
+        elif expr.operator == Binop.GEQ:
+            return f"({lhs}) <= ({rhs})"
+        elif expr.operator == Binop.GT:
+            return f"({lhs}) < ({rhs})"
         elif expr.operator == Binop.EQ:
             return f"({lhs}) = ({rhs})"
         elif expr.operator == Binop.PLUS:
@@ -142,6 +146,9 @@ def expression_prism(expr: Expr) -> str:
             return f"({lhs}) - ({rhs})"
         elif expr.operator == Binop.TIMES:
             return f"({lhs}) * ({rhs})"
+        elif expr.operator == Binop.MODULO:
+            return f"({lhs}) * ({rhs})"
+        raise PrismTranslatorException("Operator not implemented:", expr)
     elif isinstance(expr, SubstExpr):
         raise PrismTranslatorException(
             "Substitution expression not implemented:", expr)