Fixing tests

marcoeilers · Jan 25, 2024 · 12f52b9 · 12f52b9
1 parent fabb8dc
commit 12f52b9
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/src/nagini_translation/analyzer.py b/src/nagini_translation/analyzer.py
@@ -674,6 +674,8 @@ def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
             func.method_type = MethodType.class_method
             self.current_class._has_classmethod = True
         func.predicate = self.is_predicate(node)
+        if func.predicate:
+            func.contract_only = self.is_declared_contract_only(node)
         if self.is_all_low(node):
             self.has_all_low = True
             func.all_low = True

diff --git a/tests/functional/translation/test_abstract_pred_2.py b/tests/functional/translation/test_abstract_pred_2.py
@@ -13,4 +13,5 @@ def huhFunc(mc: MyClass) -> int:
     Requires(huh(mc))
     Ensures(Result() > 0)
     #:: ExpectedOutput(invalid.program:abstract.predicate.fold)
-    return Unfold(huh(mc))
+    Unfold(huh(mc))
+    return 3
diff --git a/tests/functional/translation/test_abstract_pred_3.py b/tests/functional/translation/test_abstract_pred_3.py
@@ -12,4 +12,5 @@ def huh(mc: MyClass) -> bool:
 def huhFunc(mc: MyClass) -> int:
     Ensures(Result() > 0)
     #:: ExpectedOutput(invalid.program:abstract.predicate.fold)
-    return Fold(huh(mc))
+    Fold(huh(mc))
+    return 4
diff --git a/tests/functional/translation/test_abstract_pred_4.py b/tests/functional/translation/test_abstract_pred_4.py
@@ -13,4 +13,5 @@ def huh(self) -> bool:
 def huhFunc(mc: MyClass) -> int:
     Ensures(Result() > 0)
     #:: ExpectedOutput(invalid.program:abstract.predicate.fold)
-    return Fold(mc.huh())
+    Fold(mc.huh())
+    return 4