Fix modulo, add power and divide

Philipp15b · Sep 24, 2023 · dacae69 · dacae69
1 parent c6eb165
commit dacae69
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/probably/prism/translate.py b/probably/prism/translate.py
@@ -147,7 +147,11 @@ def expression_prism(expr: Expr) -> str:
         elif expr.operator == Binop.TIMES:
             return f"({lhs}) * ({rhs})"
         elif expr.operator == Binop.MODULO:
-            return f"({lhs}) * ({rhs})"
+            return f"mod({lhs}, {rhs})"
+        elif expr.operator == Binop.POWER:
+            return f"pow({lhs}, {rhs})"
+        elif expr.operator == Binop.DIVIDE:
+            return f"pow({lhs}, {rhs})"
         raise PrismTranslatorException("Operator not implemented:", expr)
     elif isinstance(expr, SubstExpr):
         raise PrismTranslatorException(