From d15892b7b527e0a497bfa2523784e151a2970388 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Fri, 17 Nov 2023 16:33:39 +0800
Subject: [PATCH] fix: remove test foldre

---
 src/{testcode => }/demo2.py                   |  0
 src/{testcode => }/lenth_test.py              |  0
 .../speed_test.py => maxnode_test.py}         |  0
 src/{testcode => }/perf.log                   |  0
 src/speed_test.py                             | 61 +++++++++++++++++++
 5 files changed, 61 insertions(+)
 rename src/{testcode => }/demo2.py (100%)
 rename src/{testcode => }/lenth_test.py (100%)
 rename src/{testcode/speed_test.py => maxnode_test.py} (100%)
 rename src/{testcode => }/perf.log (100%)
 create mode 100644 src/speed_test.py

diff --git a/src/testcode/demo2.py b/src/demo2.py
similarity index 100%
rename from src/testcode/demo2.py
rename to src/demo2.py
diff --git a/src/testcode/lenth_test.py b/src/lenth_test.py
similarity index 100%
rename from src/testcode/lenth_test.py
rename to src/lenth_test.py
diff --git a/src/testcode/speed_test.py b/src/maxnode_test.py
similarity index 100%
rename from src/testcode/speed_test.py
rename to src/maxnode_test.py
diff --git a/src/testcode/perf.log b/src/perf.log
similarity index 100%
rename from src/testcode/perf.log
rename to src/perf.log
diff --git a/src/speed_test.py b/src/speed_test.py
new file mode 100644
index 0000000..3762a93
--- /dev/null
+++ b/src/speed_test.py
@@ -0,0 +1,61 @@
+from tpre import *
+import time
+
+N = 20
+T = N // 2
+print(f"当前门限值: N = {N}, T = {T}")
+
+total_time = 0
+
+# 1
+start_time = time.time()
+pk_a, sk_a = GenerateKeyPair()
+m = b"hello world"
+end_time = time.time()
+elapsed_time = end_time - start_time
+total_time += elapsed_time
+print(f"密钥生成运行时间:{elapsed_time}秒")
+
+# 2
+start_time = time.time()
+capsule_ct = Encrypt(pk_a, m)
+end_time = time.time()
+elapsed_time = end_time - start_time
+total_time += elapsed_time
+print(f"加密算法运行时间:{elapsed_time}秒")
+
+# 3
+pk_b, sk_b = GenerateKeyPair()
+
+# 5
+start_time = time.time()
+id_tuple = tuple(range(N))
+rekeys = GenerateReKey(sk_a, pk_b, N, T, id_tuple)
+end_time = time.time()
+elapsed_time = end_time - start_time
+total_time += elapsed_time
+print(f"重加密密钥生成算法运行时间:{elapsed_time}秒")
+
+# 7
+start_time = time.time()
+cfrag_cts = []
+
+for rekey in rekeys:
+    cfrag_ct = ReEncrypt(rekey, capsule_ct)
+    cfrag_cts.append(cfrag_ct)
+end_time = time.time()
+elapsed_time = (end_time - start_time) / len(rekeys)
+total_time += elapsed_time
+print(f"重加密算法运行时间:{elapsed_time}秒")
+
+# 9
+start_time = time.time()
+cfrags = mergecfrag(cfrag_cts)
+m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
+end_time = time.time()
+elapsed_time = end_time - start_time
+total_time += elapsed_time
+print(f"解密算法运行时间:{elapsed_time}秒")
+print("成功解密:", m)
+print(f"算法总运行时间:{total_time}秒")
+print()