|
| 1 | +import difflib |
| 2 | +import os |
| 3 | +import sys |
| 4 | + |
| 5 | + |
| 6 | +def find_diff(a, b): |
| 7 | + return "\n".join(difflib.unified_diff(a.splitlines(), b.splitlines())) |
| 8 | + |
| 9 | + |
| 10 | +sys.path.append( |
| 11 | + os.path.normpath( |
| 12 | + os.path.join(os.path.abspath(__file__), "..", "..", "..", "common") |
| 13 | + ) |
| 14 | +) |
| 15 | +from env_indigo import * # noqa |
| 16 | + |
| 17 | +indigo = Indigo() |
| 18 | +indigo.setOption("json-saving-pretty", True) |
| 19 | +indigo.setOption("molfile-saving-skip-date", True) |
| 20 | +indigo.setOption("ignore-stereochemistry-errors", True) |
| 21 | + |
| 22 | +print("*** KET to SDF ***") |
| 23 | + |
| 24 | +root = joinPathPy("molecules/", __file__) |
| 25 | +ref_path = joinPathPy("ref/", __file__) |
| 26 | + |
| 27 | +files = ["acd2d_err3"] |
| 28 | + |
| 29 | +files.sort() |
| 30 | +for filename in files: |
| 31 | + try: |
| 32 | + ket = indigo.loadMoleculeFromFile( |
| 33 | + os.path.join(root, filename + ".ket") |
| 34 | + ) |
| 35 | + except: |
| 36 | + try: |
| 37 | + ket = indigo.loadQueryMoleculeFromFile( |
| 38 | + os.path.join(root, filename + ".ket") |
| 39 | + ) |
| 40 | + except IndigoException as e: |
| 41 | + print(" %s" % (getIndigoExceptionText(e))) |
| 42 | + |
| 43 | + buffer = indigo.writeBuffer() |
| 44 | + sdfSaver = indigo.createSaver(buffer, "sdf") |
| 45 | + for frag in ket.iterateComponents(): |
| 46 | + sdfSaver.append(frag.clone()) |
| 47 | + sdfSaver.close() |
| 48 | + sdf = buffer.toString() |
| 49 | + # with open(os.path.join(ref_path, filename) + ".sdf", "w") as file: |
| 50 | + # file.write(sdf) |
| 51 | + |
| 52 | + with open(os.path.join(ref_path, filename) + ".sdf", "r") as file: |
| 53 | + sdf_ref = file.read() |
| 54 | + diff = find_diff(sdf_ref, sdf) |
| 55 | + if not diff: |
| 56 | + print(filename + ".sdf:SUCCEED") |
| 57 | + else: |
| 58 | + print(filename + ".sdf:FAILED") |
| 59 | + print(diff) |
0 commit comments