Cap number of significant digits in writer

Fix integration test accordingly

Author: pablolh

opensquirrel/Writer.py

@@ -3,9 +3,22 @@
 
 
 class Writer:
+    NUMBER_OF_SIGNIFICANT_DIGITS = 8
+
     def __init__(self, gates):
         self.gates = gates
 
+    @classmethod
+    def _format_arg(cls, squirrelAST, arg, t: ArgType):
+        if t == ArgType.QUBIT:
+            return f"{squirrelAST.qubitRegisterName}[{arg}]"
+        if t == ArgType.INT:
+            return f"{int(arg)}"
+        if t == ArgType.FLOAT:
+            return f"{float(arg):.{Writer.NUMBER_OF_SIGNIFICANT_DIGITS}}"
+
+        assert False, "Unknown argument type"
+
     def process(self, squirrelAST):
         output = ""
         output += f"""version 3.0\n\nqubit[{squirrelAST.nQubits}] {squirrelAST.qubitRegisterName}\n\n"""
@@ -21,10 +34,7 @@ def process(self, squirrelAST):
             gateName, gateArgs = operation
             signature = querySignature(self.gates, gateName)
 
-            args = [
-                f"{squirrelAST.qubitRegisterName}[{arg}]" if t == ArgType.QUBIT else f"{arg}"
-                for arg, t in zip(gateArgs, signature)
-            ]
+            args = [Writer._format_arg(squirrelAST, arg, t) for arg, t in zip(gateArgs, signature)]
 
             output += f"{gateName} {', '.join(args)}\n"
 

test/test_integration.py

@@ -48,26 +48,26 @@ def test_simple(self):
 
 qubit[3] qreg
 
-rz qreg[0], 3.141592653589793
+rz qreg[0], 3.1415927
 x90 qreg[0]
-rz qreg[0], 1.9115926535897927
+rz qreg[0], 1.9115927
 x90 qreg[0]
-rz qreg[1], 3.141592653589793
+rz qreg[1], 3.1415927
 x90 qreg[1]
-rz qreg[1], 2.37238898038469
+rz qreg[1], 2.372389
 x90 qreg[1]
-rz qreg[1], 3.141592653589793
+rz qreg[1], 3.1415927
 cz qreg[0], qreg[1]
-rz qreg[1], 3.141592653589793
+rz qreg[1], 3.1415927
 x90 qreg[1]
-rz qreg[1], 1.57238898038469
+rz qreg[1], 1.572389
 x90 qreg[1]
-rz qreg[1], 3.141592653589793
-rz qreg[0], 1.5707963267948966
+rz qreg[1], 3.1415927
+rz qreg[0], 1.5707963
 x90 qreg[0]
-rz qreg[0], 0.8415926535897933
+rz qreg[0], 0.84159265
 x90 qreg[0]
-rz qreg[0], 1.5707963267948966
+rz qreg[0], 1.5707963
 """,
         )
 
@@ -91,15 +91,15 @@ def test_qi(self):
             H q[2]
             H q[1]
             H q[0]
-            RZ q[0], 1.5707963267949
+            RZ q[0], 1.5707963
             RY q[0], -0.2
             cnot q[1], q[0]
             RZ q[0], 1.5789
             cnot q[1], q[0]
             cnot q[1], q[2]
-            RZ q[1], 2.5707963267949
+            RZ q[1], 2.5707963
             cr q[2], q[3], 2.123
-            RY q[1], -1.5707963267949
+            RY q[1], -1.5707963
 
             """,
         )
@@ -112,30 +112,30 @@ def test_qi(self):
 qubit[4] q
 
 x90 q[1]
-rz q[1], 1.5707963267948966
+rz q[1], 1.5707963
 x90 q[1]
-rz q[0], -0.20000000000000018
+rz q[0], -0.2
 x90 q[0]
-rz q[0], 1.5707963267948957
+rz q[0], 1.5707963
 x90 q[0]
-rz q[0], 1.5707963267949
+rz q[0], 1.5707963
 cnot q[1], q[0]
-rz q[0], -2.352142653589793
+rz q[0], -2.3521427
 x90 q[0]
-rz q[0], 3.141592653589793
+rz q[0], 3.1415927
 x90 q[0]
-rz q[0], 0.7894500000000004
+rz q[0], 0.78945
 cnot q[1], q[0]
 x90 q[2]
-rz q[2], 1.5707963267948966
+rz q[2], 1.5707963
 x90 q[2]
 cnot q[1], q[2]
 cr q[2], q[3], 2.123
-rz q[1], 2.5707963267949
+rz q[1], 2.5707963
 x90 q[1]
-rz q[1], 1.570796326794893
+rz q[1], 1.5707964
 x90 q[1]
-rz q[1], 3.141592653589793
+rz q[1], 3.1415927
 """
 
         self.assertEqual(output, expected)