From 770d7a560bb6d99b85cea603cadd0696b0880005 Mon Sep 17 00:00:00 2001
From: Aliaksandr Dziarkach
 <18146690+AliaksandrDziarkach@users.noreply.github.com>
Date: Thu, 20 Feb 2025 12:41:11 +0300
Subject: [PATCH] fix ut

---
 api/tests/integration/tests/formats/ket_to_ket.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/api/tests/integration/tests/formats/ket_to_ket.py b/api/tests/integration/tests/formats/ket_to_ket.py
index c5128bd33e..a0183d9b6b 100644
--- a/api/tests/integration/tests/formats/ket_to_ket.py
+++ b/api/tests/integration/tests/formats/ket_to_ket.py
@@ -60,16 +60,16 @@ def check_res(filename, format, ket_ref, ket):
     "monomer_shape",
     "ambiguous_monomer",
 ]
-formats = {
+savers = {
     "doc": [indigo.loadKetDocument],
     "mol": [indigo.loadMolecule, indigo.loadQueryMolecule],
 }
 for filename in sorted(files):
-    for format in sorted(formats.keys()):
+    for format in sorted(savers.keys()):
         file_path = os.path.join(ref_path, filename)
         with open("{}_{}.ket".format(file_path, format), "r") as file:
             ket_ref = file.read()
-        for loader in formats[format]:
+        for loader in savers[format]:
             mol = loader(ket_ref)
             # with open("{}_{}.ket".format(file_path, format), "w") as file:
             #     file.write(mol.json())
@@ -79,8 +79,9 @@ def check_res(filename, format, ket_ref, ket):
 filename = "2707_subst_count"
 file_path = os.path.join(ref_path, filename)
 mol = indigo.loadQueryMoleculeFromFile("{}.ket".format(file_path))
-for format, saver in {"ket": mol.json, "mol": mol.molfile}.items():
-    data = saver()
+savers = {"ket": mol.json, "mol": mol.molfile}
+for format in sorted(savers.keys()):
+    data = savers[format]()
     # with open("{}.{}".format(file_path, format), "w") as file:
     #     file.write(data)
     with open("{}.{}".format(file_path, format), "r") as file: