diff --git a/test_CoreAlgo.py b/test_CoreAlgo.py
index 4b38031..ab5f302 100644
--- a/test_CoreAlgo.py
+++ b/test_CoreAlgo.py
@@ -1,22 +1,14 @@
 import unittest
 import CoreAlgorithm as paillier
-from typing import Tuple
 
 
-def generate_paillier_keypair() -> (
-    Tuple[paillier.PaillierPublicKey, paillier.PaillierPrivateKey]
-):
-    """生成Paillier公钥和私钥对."""
-    public_key, private_key = paillier.generate_paillier_keypair()
-    return public_key, private_key
-
 
 class TestPaillierEncryption(unittest.TestCase):
     """测试Paillier加密算法的不同运算功能."""
 
     @classmethod
     def setUpClass(cls):
-        cls.public_key, cls.private_key = generate_paillier_keypair()
+        cls.public_key, cls.private_key = paillier.generate_paillier_keypair()
         cls.x = 90000.23
         cls.y = 90
         cls.z = 0.5