Merge pull request #283 from OpShin/fix/deterministic_compilation

Fix nondeterministic compilation

opshin/__main__.py

@@ -295,7 +295,7 @@ def perform_command(args):
     if command == Command.compile_pluto:
         print(code.dumps())
         return
-    code = pluthon.compile(code)
+    code = pluthon.compile(code, optimize_patterns=not args.no_optimize_patterns)
 
     # apply parameters from the command line to the contract (instantiates parameterized contract!)
     code = code.term
@@ -402,6 +402,11 @@ def parse_args():
         action="store_true",
         help="Enables the use of isinstance(x, D) in the contract where x is of type Anything. This is not recommended as it only checks the constructor id and not the actual type of the data.",
     )
+    a.add_argument(
+        "--no-optimize-patterns",
+        action="store_true",
+        help="Disables the compression of re-occurring code patterns. Can reduce memory and CPU steps but increases the size of the compiled contract.",
+    )
     a.add_argument(
         "args",
         nargs="*",

opshin/tests/test_misc.py

@@ -1,3 +1,7 @@
+import sys
+
+import subprocess
+
 import json
 import tempfile
 import xml.etree.ElementTree
@@ -2171,3 +2175,39 @@ def validator(
     return d
 """
         builder._compile(source_code)
+
+    def test_compilation_deterministic_local(self):
+        input_file = "examples/smart_contracts/assert_sum.py"
+        with open(input_file) as fp:
+            source_code = fp.read()
+        code = builder._compile(source_code)
+        for i in range(10):
+            code_2 = builder._compile(source_code)
+            self.assertEqual(code.dumps(), code_2.dumps())
+
+    def test_compilation_deterministic_external(self):
+        input_file = "examples/smart_contracts/assert_sum.py"
+        code = subprocess.run(
+            [
+                sys.executable,
+                "-m",
+                "opshin",
+                "compile",
+                "spending",
+                input_file,
+            ],
+            capture_output=True,
+        )
+        for i in range(10):
+            code_2 = subprocess.run(
+                [
+                    sys.executable,
+                    "-m",
+                    "opshin",
+                    "compile",
+                    "spending",
+                    input_file,
+                ],
+                capture_output=True,
+            )
+            self.assertEqual(code.stdout, code_2.stdout)