test tweaks

Author: sserita

loqs-v1.0/loqs/tools/dasktools.py

@@ -64,7 +64,7 @@ def run_program_list(
         program.shot_histories = []
 
         # Write program to file so that other processes can load
-        with NamedTemporaryFile("w+", suffix=".json", delete=False) as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix=".json", delete=False) as tempf:
             p_dict = program.to_serialization(ignore_no_serialize_flags=True)
             json.dump(p_dict, tempf)
             program_filenames.append(tempf.name)

loqs-v1.0/tests/backends/test_noise_integrated.py

@@ -104,8 +104,8 @@ def test_1q_depolarizing(self):
         # Because we set the seed, we should exactly match output from test creation
         # Also, because our depol rate is 10%, we expected a flip 5% of the time (the X and Y errors)
         # For 1000 shots, this is ~50 shots should flip
-        expected_outs = {0: 952, 1: 48}
-        assert Counter(outs) == expected_outs
+        assert abs(Counter(outs)[0] - 950) < 10
+        assert abs(Counter(outs)[1] - 50) < 10
 
         # Also test QuantumSim in the X basis
         stack_Xbasis = [
@@ -121,7 +121,8 @@ def test_1q_depolarizing(self):
         program_qsim_Xbasis.run(num_shots=1000, reset_shot_histories=True)
         outs = [mo["Q0"][0] for mo in program_qsim_Xbasis.collect_shot_data("measurement_outcomes", -1)]
         # Now Y and Z should flip. Because rate X == rate Z and we have RNG, results should be unchanged
-        assert Counter(outs) == expected_outs
+        assert abs(Counter(outs)[0] - 950) < 10
+        assert abs(Counter(outs)[1] - 50) < 10
 
         ## STIM
         gate_dict, inst_dict = self._create_model_dicts(qubits, GateRep.STIM_CIRCUIT_STR)
@@ -144,15 +145,17 @@ def test_1q_depolarizing(self):
         outs = [mo["Q0"][0] for mo in program_stim.collect_shot_data("measurement_outcomes", -1)]
         # STIM handles its own RNG, so this could in principle differ from QuantumSim results
         # In practice, for this simple circuit, I've found that this coincidentally matches for Z basis
-        assert Counter(outs) == expected_outs
+        assert abs(Counter(outs)[0] - 950) < 10
+        assert abs(Counter(outs)[1] - 50) < 10
 
         # Also test STIM in the X basis
         program_stim_Xbasis = QuantumProgram.from_quantum_program(program_stim, stack_Xbasis)
 
         program_stim_Xbasis.run(num_shots=1000)
         outs = [mo["Q0"][0] for mo in program_stim_Xbasis.collect_shot_data("measurement_outcomes", -1)]
         # But the RNG is handled differently for X basis, so we don't get a match (although it is still ~50)
-        assert Counter(outs) == {0: 944, 1: 56}
+        assert abs(Counter(outs)[0] - 950) < 10
+        assert abs(Counter(outs)[1] - 50) < 10
 
     # Amp damp tests, given as damping rate, dephasing rate, seed, num shots, and then expected 1 counts in order of:
     # QuantumSim X,I,Mz; QuantumSim H,I,Mz; QuantumSim H,I,Mx; STIM X,I,Mz; STIM H,I,Mz; STIM H,I,Mx

loqs-v1.0/tests/backends/test_npsvstate.py

@@ -256,7 +256,7 @@ def test_serialization(self):
         test, _ = state10.apply_reps([RepTuple(U_H, ["Q1"], GateRep.UNITARY)])
         test.apply_reps_inplace([RepTuple(U_CZ, ["Q0", "Q1"], GateRep.UNITARY)])
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             test.write(tempf.name)
 
             test2: SVState = SVState.read(tempf.name)

loqs-v1.0/tests/backends/test_qsimstate.py

@@ -257,7 +257,7 @@ def test_serialization(self):
         # Don't force propagation here
         # So serialization should both serialize DM and operations to be applied
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             test.write(tempf.name)
 
             test2: QSimState = QSimState.read(tempf.name)

loqs-v1.0/tests/backends/test_stimstate.py

@@ -219,7 +219,7 @@ def test_serialization(self):
         test = STIMState([1, 0], ["Q0", "Q1"])
         test.state.cx(0, 1)
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             test.write(tempf.name)
 
             test2 = STIMState.read(tempf.name)

loqs-v1.0/tests/core/instructions/test_instruction.py

@@ -67,7 +67,7 @@ def map_qubits_fn(qubit_mapping, qubits, **kwargs):
 
         ins = Instruction(apply_fn, data, map_qubits_fn, name="test")
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             ins.write(tempf.name)
 
             ins2 = Frame.read(tempf.name)
@@ -84,7 +84,7 @@ def map_qubits_fn(qubit_mapping, qubits, **kwargs):
         # We should be able to do it a second time also
         # This is because we cache the serialization the first time,
         # so even though the second time we don't have access to the source code, it works
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             ins2.write(tempf.name)
 
             ins3 = Frame.read(tempf.name)

loqs-v1.0/tests/core/instructions/test_instructionlabel.py

@@ -71,7 +71,7 @@ def test_serialization(self):
         # Test string version
         ilbl = InstructionLabel("Label", "L0", self.args, self.kwargs)
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             ilbl.write(tempf.name)
 
             ilbl2 = InstructionLabel.read(tempf.name)
@@ -80,7 +80,7 @@ def test_serialization(self):
         # And instruction version
         ilbl3 = InstructionLabel(self.ins, "L0", self.args, self.kwargs)
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             ilbl3.write(tempf.name)
 
             ilbl4 = InstructionLabel.read(tempf.name)

loqs-v1.0/tests/core/instructions/test_instructionstack.py

@@ -52,7 +52,7 @@ def test_list_operations(self):
     def test_serialization(self):
         s = InstructionStack([self.ilbl1, self.ilbl2]) # type: ignore
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             s.write(tempf.name)
 
             s2 = InstructionStack.read(tempf.name)

loqs-v1.0/tests/core/recordables/test_measurementoutcomes.py

@@ -53,7 +53,7 @@ def test_serialization(self):
         expected = {"Q0": [0, 1], "Q1": [1]}
         m = MeasurementOutcomes(outcomes)
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             m.write(tempf.name)
 
             m2 = MeasurementOutcomes.read(tempf.name)

loqs-v1.0/tests/core/recordables/test_patchdict.py

@@ -29,7 +29,7 @@ def test_serialization(self):
         patch2 = code.create_patch(["D1", "A1"])
         patches = PatchDict({"L0": patch1, "L1": patch2})
 
-        with NamedTemporaryFile("w+", suffix='.json') as tempf:
+        with NamedTemporaryFile("w+", dir='.', suffix='.json') as tempf:
             patches.write(tempf.name)
 
             patches2 = PatchDict.read(tempf.name)