Change test expected

KMU-quantum-classroom · Oct 12, 2023 · 1e1cb0a · 1e1cb0a
1 parent 1d915ec
commit 1e1cb0a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/qiskit_classroom/worker.py b/qiskit_classroom/worker.py
@@ -54,6 +54,7 @@ def add_new_line(strings: list[str]) -> str:
 class ConverterWorker:
     """worker for convert expression and visualize expression"""
 
+    # pylint: disable=too-many-arguments
     def __init__(
         self,
         from_expression: QuantumExpression,
@@ -174,7 +175,10 @@ def generate_visualization_code(self) -> str:
             if self.to_expression is QuantumExpression.MATRIX:
                 return add_new_line(
                     [
-                        "for gate, name in zip(reversed(result['gate']), reversed(result['name'])):",
+                        (
+                            "for gate, name in zip(reversed(result['gate']),"
+                            + "reversed(result['name'])):"
+                        ),
                         "\totimes=' \\\\otimes '",
                         """\tprint('\\stackrel{' + otimes.join(name[1]) +'}' + f'{{{gate}}}')""",
                         "print(f\"= \\stackrel{{result}}{{{result['result']}}}\")"

diff --git a/tests/test_worker.py b/tests/test_worker.py
@@ -42,8 +42,9 @@
     "converter = ConversionService(conversion_type='QC_TO_MATRIX',"
     + " option={'print': 'raw'})\n"
     + f"result = converter.convert(input_value={VALUE_NAME})",
-    "for gate, name in zip(reversed(result['gate']), reversed(result['name'])):\n"
-    + "\tprint(f'{gate.strip()}_' + '{' + \"\\\\otimes \".join(name[1]) + '}')\n",
+    "for gate, name in zip(reversed(result['gate']),reversed(result['name'])):\n"
+    + "\totimes=' \\\\otimes '\n"
+    + "\tprint('\\stackrel{' + otimes.join(name[1]) +'}' + f'{{{gate}}}')\n",
 ]
 
 MATRIX_TO_QC_EXPECTED = [