from setuptools import setup
from Cython.Build import cythonize
import numpy as np

setup(
    ext_modules=cythonize(
        "UCS_test_0330.py",          # 要编译的Python文件
        compiler_directives={
            'language_level': 3,  # Python 3
            'boundscheck': False, # 关闭边界检查提升性能
            'wraparound': False,
        }
    ),
    include_dirs=[np.get_include()],  # 包含NumPy头文件
)