Fix repetition code decoder test (#407)

* fix rep test * lintng
qiskit-community · Nov 6, 2023 · aebaa93 · aebaa93
1 parent efff153
commit aebaa93
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/qiskit_qec/circuits/repetition_code.py b/src/qiskit_qec/circuits/repetition_code.py
@@ -349,7 +349,7 @@ def string2raw_logicals(self, string):
         Returns:
             list: Raw values for logical operators that correspond to nodes.
         """
-        return _separate_string(self._process_string(string))[0]
+        return string.split(" ", maxsplit=1)[0][-1]
 
     def check_nodes(self, nodes, ignore_extra_boundary=False, minimal=False):
         """

diff --git a/test/code_circuits/test_rep_codes.py b/test/code_circuits/test_rep_codes.py
@@ -538,7 +538,7 @@ def clustering_decoder_test(
         # now run them all and check it works
         for c, code in enumerate(codes):
             decoding_graph = DecodingGraph(code)
-            if c == 3 and Decoder is UnionFindDecoder:
+            if c >= 0 and Decoder is UnionFindDecoder:
                 decoder = Decoder(code, decoding_graph=decoding_graph, use_peeling=False)
             else:
                 decoder = Decoder(code, decoding_graph=decoding_graph)
@@ -555,7 +555,7 @@ def clustering_decoder_test(
                 for j, z_logical in enumerate(decoder.measured_logicals):
                     error = corrected_z_logicals[j] != 1
                     if error:
-                        error_num = string.count("0")
+                        error_num = string.split(" ", maxsplit=1)[0].count("0")
                         if error_num < min_error_num:
                             min_error_num = error_num
                             min_error_string = string