ion-CSP 2.2.0__tar.gz → 2.2.2__tar.gz

{ion_csp-2.2.0 → ion_csp-2.2.2}/CHANGELOG.md

@@ -2,6 +2,30 @@
 
 ## Latest Changes
 
+### d914d90 (2025-07-03)
+
+Update the version to 2.2.2. The redundant output content of the console is greatly reduced to control the size of the console. log file. Specifically, the redundant log records are reduced in gen_opt when exceptions occur in the structure generation process, and the redundant log records in gen_opt and read_mlp_density are reduced when calling the phonopy toolkit.
+
+## V2.2.1
+
+### 0a77f51 (2025-07-01)
+
+Due to the unsatisfactory performance of Mattersim in testing, we switched back to fine-tuning the machine learning potential of DP.  The performance of ExpCellFilter in testing is easier to identify experimental structures compared to UnitCellFilter, so it is changed to call ExpCellFilter in mlp_opt.py.
+
+### 1754cc7 (2025-06-30)
+
+Update version number to 2.2.1, fix file copying logic in main_EE workflow, update mlp_opt.py to use MatterSimCalculator to develop calculation efficiency.
+
+### d6abfb5 (2025-06-28)
+
+Optimize the environment variable settings in mlp_opt.py to disable numpy's multithreading and ensure the stability of multi-threaded computation.
+
+## V2.2.0
+
+### 18b9eba (2025-06-28)
+
+Optimize the file copying logic in vasp_processing to ensure that copying operations are only performed when the file exists; Update sorting parameters to support 'NC_ratio' and modify the relevant CSV file names accordingly.
+
 ### 14ffac2 (2025-06-27)
 
 Update version number to 2.2.0, optimize file replication logic in VASP processing, add error handling to ensure errors are recorded when files do not exist, modify README and CHANGELOG to reflect the latest changes.

{ion_csp-2.2.0 → ion_csp-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ion_CSP
-Version: 2.2.0
+Version: 2.2.2
 Summary: Crystal Structure Design Software Based on Molecular/Ionic Configuration.
 Home-page: https://github.com/bagabaga007/ion_CSP
 Author: Ze Yang
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ase==3.23.0
 Requires-Dist: scipy==1.15.2
-Requires-Dist: deepmd-kit==3.0.1
+Requires-Dist: deepmd-kit>=3.0.1
 Requires-Dist: torch==2.5.0
 Requires-Dist: dpdispatcher==0.6.7
 Requires-Dist: numpy==1.26.4

{ion_csp-2.2.0 → ion_csp-2.2.2}/pyproject.toml

@@ -20,7 +20,7 @@ requires-python = ">=3.11"
 dependencies = [
     "ase==3.23.0",
     "scipy==1.15.2",
-    "deepmd-kit==3.0.1",
+    "deepmd-kit>=3.0.1",
     "torch==2.5.0",
     "dpdispatcher==0.6.7",
     "numpy==1.26.4",

{ion_csp-2.2.0 → ion_csp-2.2.2}/requirements.txt

@@ -1,6 +1,6 @@
 ase==3.23.0
 scipy==1.15.2
-deepmd-kit==3.0.1
+deepmd-kit>=3.0.1
 torch==2.5.0
 dpdispatcher>=0.6.7
 numpy==1.26.4

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/__init__.py

@@ -1,8 +1,8 @@
 __author__ = "Ze Yang"
 __contact__ = "yangze1995007@163.com"
 __license__ = "MIT"
-__version__ = "2.2.0"
-__date__ = "2025-06-27"
+__version__ = "2.2.2"
+__date__ = "2025-07-02"
 
 
 try:

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/convert_SMILES.py

@@ -131,7 +131,7 @@ class SmilesProcessing:
                 f"{self.converted_dir}/charge_{charge}"
             )
             os.makedirs(charge_dir, exist_ok=True)
-            # 通过SMILE_to函数依次处理SMILES码
+            # 通过_convert_SMILES函数依次处理SMILES码
             for _, row in group.iterrows():
                 result_code, basename = self._convert_SMILES(
                     dir=charge_dir,
@@ -238,8 +238,10 @@ class SmilesProcessing:
         for folder in folders:
             folder_dir = os.path.join(self.converted_dir, folder)
             if not os.path.exists(folder_dir):
-                logging.error(f'Provided folder {folder} is not in the directory {folder_dir}')
-                continue
+                folder_dir = os.path.join(self.base_dir, folder)
+                if not os.path.exists(folder_dir):
+                    logging.error(f'Provided folder {folder} is not either in the work directory or the converted directory.\n')
+                    continue
             # 获取文件夹中所有以 .gjf 结尾的文件
             gjf_files = [
                 f for f in os.listdir(folder_dir) if f.endswith(".gjf")

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/gen_opt.py

@@ -116,9 +116,8 @@ class CrystalGenerator:
                     pyxtal_structure.to_file(POSCAR_path, fmt="poscar")
                     total_count += 1
                     group_count += 1
+                # 捕获对于某一空间群生成结构的运行时间过长、组成兼容性错误、对称性兼容性错误等异常，使结构生成能够完全进行而不中断
                 except (RuntimeError, Comp_CompatibilityError, Symm_CompatibilityError) as e:
-                    # 捕获对于某一空间群生成结构的运行时间过长、组成兼容性错误、对称性兼容性错误等异常，使结构生成能够完全进行而不中断
-                    logging.error(f"Generating structure error: {e}")
                     # 记录异常类型并跳出当前空间群的生成循环
                     exception_message = type(e).__name__
                     break
@@ -154,7 +153,11 @@ class CrystalGenerator:
         # 按顺序处理POSCAR文件，首先复制一份无数字后缀的POSCAR文件
         shutil.copy(f"{self.POSCAR_dir}/{filename}", f"{self.POSCAR_dir}/POSCAR")
         try:
-            subprocess.run(["nohup", "phonopy", "--symmetry", "POSCAR"], check=True)
+            subprocess.run(
+                ["nohup", "phonopy", "--symmetry", "POSCAR"],
+                check=True,
+                stdout=subprocess.DEVNULL
+            )
         except subprocess.CalledProcessError as e:
             # 新增：捕获phonopy执行错误
             logging.error(f"Phonopy execution failed for {filename}: {str(e)}")
@@ -168,9 +171,6 @@ class CrystalGenerator:
         
         # 检查生成的POSCAR中的原子数，如果不匹配则删除该POSCAR并在日志中记录
         if cell_atoms != self.cell_atoms:
-            error_message = f"Atom number mismatch ({cell_atoms} vs {self.cell_atoms})"
-            print(f"{filename} - {error_message}")
-            
             # 新增：回溯空间群归属
             poscar_index = int(filename.split('_')[1])  # 提取POSCAR编号
             space_group = self._find_space_group(poscar_index)

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/mlp_opt.py

@@ -1,23 +1,27 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 import os
+os.environ["MKL_NUM_THREADS"] = "1"
+os.environ["NUMEXPR_NUM_THREADS"] = "1"
+os.environ["OMP_NUM_THREADS"] = "1"
+
 import time
 import numpy as np
 import multiprocessing
 from ase.io.vasp import read_vasp
 from ase.optimize import LBFGS
-from ase.constraints import UnitCellFilter
+# from ase.constraints import UnitCellFilter
+from ase.filters import ExpCellFilter
 from deepmd.calculator import DP
+# from mattersim.forcefield import MatterSimCalculator
 
 # 根据脚本位置确定model.pt文件的位置, 减少错误发生
 base_dir = os.path.dirname(__file__)
 relative_path = './model.pt'
 file_path = os.path.join(base_dir, relative_path)
 calc = DP(file_path)
-"""
-structure optimization with DP model and ASE
-PSTRESS and fmax should exist in input.dat
-"""
+# calc = MatterSimCalculator(device="cuda")
+
 
 def get_element_num(elements):
     """
@@ -51,7 +55,7 @@ def write_CONTCAR(element, ele, lat, pos, index):
         pos: atomic positions in direct coordinates
         index: index for the output file""" 
     f = open(f'{base_dir}/CONTCAR_'+str(index),'w')
-    f.write('ASE-DPKit-Optimization\n')
+    f.write('ASE-MLP-Optimization\n')
     f.write('1.0\n') 
     for i in range(3):
         f.write('%15.10f %15.10f %15.10f\n' % tuple(lat[i]))
@@ -139,7 +143,7 @@ def get_indexes():
 
 def run_opt(index: int): 
     """
-    Using the ASE&DP to Optimize Configures
+    Using the ASE & MLP to Optimize Configures
     :params
         index: index of the POSCAR file to be optimized
     """
@@ -147,7 +151,7 @@ def run_opt(index: int):
         os.system(f'mv {base_dir}/OUTCAR {base_dir}/OUTCAR-last')
     fmax, pstress = 0.03, 0
 
-    print('Start to Optimize Structures by DP----------')
+    print('Start to Optimize Structures by MLP----------')
         
     Opt_Step = 2000
     start = time.time() 
@@ -158,7 +162,8 @@ def run_opt(index: int):
     aim_stress = 1.0 * pstress * 0.01 * 0.6242 / 10.0 
     atoms = read_vasp('POSCAR_'+str(index)) 
     atoms.calc = calc 
-    ucf = UnitCellFilter(atoms, scalar_pressure=aim_stress)
+    # ucf = UnitCellFilter(atoms, scalar_pressure=aim_stress)
+    ucf = ExpCellFilter(atoms, scalar_pressure=aim_stress)
     # optimization
     opt = LBFGS(ucf) 
     opt.run(fmax=fmax,steps=Opt_Step) 

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/read_mlp_density.py

@@ -8,7 +8,7 @@ from ion_CSP.identify_molecules import identify_molecules, molecules_information
 
 class ReadMlpDensity:
     
-    def __init__(self, work_dir:str):
+    def __init__(self, work_dir:str, folder:str = '2_mlp_optimized'):
         """
         This class is designed to read and process MLP optimized files, specifically CONTCAR files, to calculate and sort their densities.
         The class also provides functionality to process these files using phonopy for symmetry analysis and primitive cell generation.
@@ -20,7 +20,7 @@ class ReadMlpDensity:
         self.base_dir = work_dir
         os.chdir(self.base_dir)
         # 寻找同一目录下的2_mlp_optimized文件夹
-        self.folder_dir = os.path.join(self.base_dir, '2_mlp_optimized')
+        self.folder_dir = os.path.join(self.base_dir, folder)
         self.max_density_dir = os.path.join(self.folder_dir, 'max_density')
         self.primitive_cell_dir = os.path.join(self.folder_dir, 'primitive_cell')
         print(f"Processing MLP CONTCARs in {self.folder_dir}")
@@ -174,7 +174,11 @@ class ReadMlpDensity:
                     # 使用phonopy模块处理POSCAR结构文件，获取对称化的原胞和常规胞。
                     # 应用晶体的对称操作优化后的原胞可以最好地符合晶体的对称性，减少后续优化计算的复杂性。
                     log.write(f'\nProcessing file: {new_CONTCAR_filename}\n')
-                    result = subprocess.run(['nohup', 'phonopy', '--symmetry', 'POSCAR'], stderr=subprocess.STDOUT)
+                    result = subprocess.run(
+                        ["nohup", "phonopy", "--symmetry", "POSCAR"],
+                        check=True,
+                        stdout=subprocess.DEVNULL
+                    )
                     log.write(f'Finished processing file: {new_CONTCAR_filename} with return code: {result.returncode}\n')
                 # 将phonopy生成的PPOSCAR（对称化原胞）和BPOSCAR（对称化常规胞）放到对应的文件夹中，并将文件名改回POSCAR_index
                 shutil.move(f'{self.phonopy_dir}/PPOSCAR', f'{self.primitive_cell_dir}/{new_CONTCAR_filename}')

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/run/main_EE.py

@@ -109,6 +109,8 @@ def combination_task(work_dir, config):
     elif config["empirical_estimate"]["sort_by"] == "nitrogen":
         # 最终将预测的离子晶体氮含量以及对应的组分输出到 .csv 文件并根据氮含量从大到小排序
         combination.nitrogen_content_estimate()
+    elif config["empirical_estimate"]["sort_by"] == "NC_ratio":
+        combination.carbon_nitrogen_ratio_estimate()
     # 基于排序依据 sort_by 对应的 .csv 文件创建 combo_n 文件夹，并复制相应的 .gjf 结构文件。
     if config["empirical_estimate"]["make_combo_dir"]:
         combination.make_combo_dir(

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP/task_manager.py

@@ -219,7 +219,9 @@ class TaskManager:
     def task_runner(self, module: str, work_dir: str):
         """任务执行器 - Task execution handler"""
         work_dir = Path(work_dir)
-        work_dir.mkdir(exist_ok=True)
+        if not os.path.exists(work_dir):
+            print(f"Work directory {work_dir} does not exist")
+            return
 
         console_log = work_dir / f"main_{module}_console.log"
         pid_file = work_dir / "pid.txt"
@@ -262,7 +264,7 @@ class TaskManager:
             std_log.symlink_to(output_log)
 
         print('Starting task ......')
-        time.sleep(6)
+        time.sleep(3)
         logging.info(f"Started {module} module (PID: {process.pid})")
         print(f"Task started (PID: {process.pid})")
         print(f"Normalized log file: {std_log}")

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ion_CSP
-Version: 2.2.0
+Version: 2.2.2
 Summary: Crystal Structure Design Software Based on Molecular/Ionic Configuration.
 Home-page: https://github.com/bagabaga007/ion_CSP
 Author: Ze Yang
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ase==3.23.0
 Requires-Dist: scipy==1.15.2
-Requires-Dist: deepmd-kit==3.0.1
+Requires-Dist: deepmd-kit>=3.0.1
 Requires-Dist: torch==2.5.0
 Requires-Dist: dpdispatcher==0.6.7
 Requires-Dist: numpy==1.26.4

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP.egg-info/SOURCES.txt

@@ -100,6 +100,6 @@ src/ion_CSP/run/run_read_mlp_density.py
 src/ion_CSP/run/run_upload_download.py
 src/ion_CSP/run/run_vasp_processing.py
 tests/__init__.py
-tests/test1_task_manager1.py
+tests/test_identify_molecules.py
 tests/test_log_and_time.py
 tests/test_task_manager.py

{ion_csp-2.2.0 → ion_csp-2.2.2}/src/ion_CSP.egg-info/requires.txt

@@ -1,6 +1,6 @@
 ase==3.23.0
 scipy==1.15.2
-deepmd-kit==3.0.1
+deepmd-kit>=3.0.1
 torch==2.5.0
 dpdispatcher==0.6.7
 numpy==1.26.4

ion_csp-2.2.2/tests/test_identify_molecules.py

@@ -0,0 +1,171 @@
+import os
+import logging
+import pytest
+from ase import Atoms
+from ion_CSP.identify_molecules import identify_molecules, molecules_information
+
+# 配置日志
+logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
+
+
+def create_gjf_file(path: str, filename: str, atoms: Atoms):
+    """手动创建临时的GJF文件"""
+    filepath = os.path.join(path, filename)
+    with open(filepath, "w") as f:
+        f.write("#N\n\n")  # Gaussian 文件头
+        f.write("Generated by ASE test\n\n")
+        f.write("0 1\n")
+        for atom in atoms:
+            f.write(f"{atom.symbol} {atom.x} {atom.y} {atom.z}\n")
+        f.write("\n\n")  # 文件结束
+    return filepath
+
+
+@pytest.fixture
+def setup_test_environment(tmp_path):
+    """设置测试环境，创建临时目录并切换工作目录"""
+    os.chdir(tmp_path)
+    return tmp_path
+
+
+def test_single_water_molecule(setup_test_environment):
+    """测试单个水分子"""
+    # 创建初始GJF文件
+    water = Atoms("OH2", positions=[[0, 0, 0], [1, 0, 0], [0, 1, 0]])
+    create_gjf_file(setup_test_environment, "water.gjf", water)
+
+    # 运行识别函数
+    merged_molecules, molecules_flag, initial_info = identify_molecules(water)
+
+    # 验证结果
+    expected_molecule = frozenset([("O", 1), ("H", 2)])
+    assert len(merged_molecules) == 1
+    assert expected_molecule in merged_molecules
+    assert merged_molecules[expected_molecule] == 1
+    assert molecules_flag is True
+    assert initial_info == [{"O": 1, "H": 2}]
+
+
+def test_two_isolated_water_molecules(setup_test_environment):
+    """测试两个独立的水分子"""
+    # 创建两个初始GJF文件
+    water1 = Atoms("OH2", positions=[[0, 0, 0], [1, 0, 0], [0, 1, 0]])
+    water2 = Atoms("OH2", positions=[[10, 0, 0], [11, 0, 0], [10, 1, 0]])
+    create_gjf_file(setup_test_environment, "water1.gjf", water1)
+    create_gjf_file(setup_test_environment, "water2.gjf", water2)
+
+    # 合并两个分子
+    combined = water1 + water2
+
+    # 运行识别函数
+    merged_molecules, molecules_flag, initial_info = identify_molecules(combined)
+
+    # 验证结果
+    expected_molecule = frozenset([("O", 1), ("H", 2)])
+    assert len(merged_molecules) == 1
+    assert expected_molecule in merged_molecules
+    assert merged_molecules[expected_molecule] == 2
+    assert molecules_flag is True
+
+
+def test_water_and_methane_molecules(setup_test_environment):
+    """测试水分子和甲烷分子"""
+    # 创建初始GJF文件
+    water = Atoms("OH2", positions=[[0, 0, 0], [1, 0, 0], [0, 1, 0]])
+    methane = Atoms(
+        "CH4", positions=[[10, 0, 0], [11, 0, 0], [10, 1, 0], [10, 0, 1], [9, 0, 0]]
+    )
+    create_gjf_file(setup_test_environment, "water.gjf", water)
+    create_gjf_file(setup_test_environment, "methane.gjf", methane)
+
+    # 合并两个分子
+    combined = water + methane
+
+    # 运行识别函数
+    merged_molecules, molecules_flag, initial_info = identify_molecules(combined)
+
+    # 验证结果
+    water_mol = frozenset([("O", 1), ("H", 2)])
+    methane_mol = frozenset([("C", 1), ("H", 4)])
+
+    assert len(merged_molecules) == 2
+    assert water_mol in merged_molecules
+    assert methane_mol in merged_molecules
+    assert merged_molecules[water_mol] == 1
+    assert merged_molecules[methane_mol] == 1
+    assert molecules_flag is True
+    assert sorted(initial_info, key=lambda x: sorted(x.items())) == sorted(
+        [{"O": 1, "H": 2}, {"C": 1, "H": 4}], key=lambda x: sorted(x.items())
+    )
+
+
+def test_mismatched_molecules(setup_test_environment):
+    """测试分子不匹配的情况"""
+    # 只创建一个初始GJF文件
+    water = Atoms("OH2", positions=[[0, 0, 0], [1, 0, 0], [0, 1, 0]])
+    create_gjf_file(setup_test_environment, "water.gjf", water)
+
+    # 创建两个独立的水分子
+    water1 = Atoms("OH", positions=[[0, 0, 0], [1, 0, 0]])
+    water2 = Atoms("OH3", positions=[[10, 0, 0], [11, 0, 0], [10, 1, 0], [10, 1, 1]])
+    combined = water1 + water2
+
+    # 运行识别函数
+    merged_molecules, molecules_flag, initial_info = identify_molecules(combined)
+
+    # 验证结果
+    expected_molecule = frozenset([("O", 1), ("H", 2)])
+    assert len(merged_molecules) == 2
+    assert expected_molecule not in merged_molecules
+    assert merged_molecules[expected_molecule] == 0
+    assert molecules_flag is False
+    assert initial_info == [{"O": 1, "H": 2}]
+
+
+def test_no_gjf_files(setup_test_environment):
+    """测试没有GJF文件的情况"""
+    water = Atoms("OH2", positions=[[0, 0, 0], [1, 0, 0], [0, 1, 0]])
+
+    # 运行识别函数
+    merged_molecules, molecules_flag, initial_info = identify_molecules(water)
+
+    # 验证结果
+    expected_molecule = frozenset([("O", 1), ("H", 2)])
+    assert len(merged_molecules) == 1
+    assert expected_molecule in merged_molecules
+    assert merged_molecules[expected_molecule] == 1
+    assert molecules_flag is False  # 没有初始GJF文件
+    assert initial_info == []
+
+
+def test_molecules_information_logging(caplog):
+    """测试分子信息日志输出"""
+    # 准备测试数据
+    molecules = {frozenset([("C", 1), ("H", 4)]): 1, frozenset([("O", 1), ("H", 2)]): 2}
+    molecules_flag = True
+    initial_info = [{"C": 1, "H": 4}, {"O": 1, "H": 2}]
+
+    # 运行信息输出函数
+    with caplog.at_level(logging.INFO):
+        molecules_information(molecules, molecules_flag, initial_info)
+
+    # 验证日志输出
+    assert "Initial molecules: [{'C': 1, 'H': 4}, {'O': 1, 'H': 2}]" in caplog.text
+    assert "Molecule 1 (Total Atoms: 5, Count: 1): C1H4" in caplog.text
+    assert "Molecule 2 (Total Atoms: 3, Count: 2): O1H2" in caplog.text
+    assert "Molecular Comparison Successful" in caplog.text
+
+
+def test_molecules_information_failed_logging(caplog):
+    """测试分子比较失败的日志输出"""
+    # 准备测试数据
+    molecules = {frozenset([("C", 1), ("H", 4)]): 1, frozenset([("O", 1), ("H", 2)]): 2}
+    molecules_flag = False
+    initial_info = [{"C": 1, "H": 4}, {"O": 1, "H": 2}]
+
+    # 运行信息输出函数
+    with caplog.at_level(logging.INFO):
+        molecules_information(molecules, molecules_flag, initial_info)
+
+    # 验证日志输出
+    assert "Molecular Comparison Failed" in caplog.text

ion_csp-2.2.0/tests/test1_task_manager1.py

@@ -1,121 +0,0 @@
-import os
-import time
-import pytest
-from pathlib import Path
-from unittest.mock import patch, MagicMock, Mock
-from ion_CSP.task_manager import TaskManager
-
-
-@pytest.fixture(scope="session", autouse=True)
-def set_working_directory():
-    """设置工作目录为项目根目录"""
-    project_root = Path(__file__).resolve().parent  # 假设测试文件在 tests 目录下
-    os.chdir(project_root)
-    yield
-
-@pytest.fixture
-def task_manager(tmp_path):
-    # 使用临时目录替代工作目录，避免污染真实文件系统
-    tm = TaskManager()
-    tm.workspace = tmp_path
-    tm.log_dir = tmp_path / tm.log_base
-    tm.log_dir.mkdir(parents=True, exist_ok=True)
-    return tm
-
-
-def test_task_runner_creates_log_and_pid(monkeypatch, tmp_path, task_manager):
-    # 模拟 subprocess.Popen，避免真正启动子进程
-    mock_process = MagicMock()
-    mock_process.pid = 9999
-
-    def fake_popen(*args, **kwargs):
-        return mock_process
-
-    monkeypatch.setattr("subprocess.Popen", fake_popen)
-
-    # 调用 task_runner
-    work_dir = tmp_path / "work"
-    task_manager.task_runner("CSP", str(work_dir))
-
-    # 检查日志目录下是否生成了符号链接（symlink）
-    symlink_path = task_manager.log_dir / f"CSP_{mock_process.pid}.log"
-    assert symlink_path.exists()
-    assert symlink_path.is_symlink()
-
-
-def test_get_related_tasks_basic(task_manager, monkeypatch):
-    """测试标准日志文件处理"""
-    # 验证 log_dir 是 Path 对象
-    assert isinstance(task_manager.log_dir, Path), "log_dir 必须是 pathlib.Path 对象"
-    log_dir = task_manager.log_dir
-    log_dir.mkdir(parents=True, exist_ok=True)
-    print(log_dir)
-    # 模拟的时间戳
-    now = time.time()
-
-    # 创建模拟的 stat 对象
-    mock_stats = {
-        8: Mock(st_mtime=now - 8000),
-        5: Mock(st_mtime=now - 5000),
-        3: Mock(st_mtime=now - 3000),
-        1: Mock(st_mtime=now - 1000),
-    }
-
-    # 模拟 os.stat 行为
-    def mock_os_stat(path, **kwargs):
-        try:
-            base_name = path.name.split(".")[0]
-            pid = int(base_name.split("_")[-1])
-            return mock_stats.get(pid, Mock(st_mtime=0))
-        except (IndexError, ValueError):
-            return Mock(st_mtime=0)
-
-    monkeypatch.setattr("os.stat", mock_os_stat)
-    monkeypatch.setattr(task_manager, "_is_pid_running", lambda _: True)
-
-    # 创建日志文件
-    for pid in [8, 5, 3, 1]:
-        log_file = log_dir / f"CSP_{pid:04d}.log"
-        log_file.touch()
-
-    # 调用方法
-    tasks = task_manager.get_related_tasks()
-    print(tasks)
-    # 断言
-    assert len(tasks) == 4
-    for task in tasks:
-        assert task["pid"] in [8, 5, 3, 1]
-        assert task["status"] == "Running"
-
-
-def test_main_menu_navigation(monkeypatch, task_manager):
-    """测试主菜单导航"""
-    monkeypatch.setattr("builtins.input", MagicMock(side_effect=["3", "q"]))
-    with patch("sys.stdout") as mock_stdout:
-        task_manager.main_menu()
-        output = mock_stdout.getvalue()
-        assert "View Logs" in output
-        assert "Terminate Tasks" in output
-
-
-# def test_get_related_tasks_process_status(task_manager, monkeypatch):
-#     """测试进程状态判断"""
-#     test_file = task_manager.log_dir / "CSP_0001.log"
-#     test_file.touch()
-
-#     # 模拟 os.stat
-#     mock_stat = Mock(st_mtime=time.time())
-#     monkeypatch.setattr("os.stat", lambda _: mock_stat)
-
-#     # 模拟进程状态
-#     monkeypatch.setattr(task_manager, "_is_pid_running", lambda pid: pid % 2 == 0)
-
-#     # 执行测试
-#     tasks = task_manager.get_related_tasks()
-
-#     # 验证结果
-#     assert len(tasks) == 1
-#     assert tasks[0]["status"] == "Terminated"  # PID=1 为奇数，应标记为终止
-
-
-# 你可以继续添加针对 _safe_kill、_is_pid_running 等方法的单元测试