chore: add benchmark and readme (FederatedAI#3859)

Signed-off-by: weiwee <[email protected]>

.github/workflows/build_wheel.yml

@@ -81,7 +81,7 @@ jobs:
       - name: Test Install
         run: |
           pip install -U pip
-          pip install --find-link=. fate_crypto
+          pip install --find-link=dist/ fate_crypto
       - name: Upload to PyPI Test
         if: ${{ github.event.inputs.type == 'testpypi' }}
         run: |

rust/fate_crypto/README.md

@@ -0,0 +1,4 @@
+# fate_crypto
+
+This package contains some crypto algorithms implemented or wrapped
+using Rust. It's for FATE use only!

rust/fate_crypto/pyproject.toml

@@ -4,10 +4,10 @@ build-backend = "maturin"
 
 [project]
 name = "fate_crypto"
+description = "crypto implemented or wrapped using rust"
+readme = "README.md"
 requires-python = ">=3.6"
 classifiers = [
     "Programming Language :: Rust",
     "Programming Language :: Python :: Implementation :: CPython",
-    "Programming Language :: Python :: Implementation :: PyPy",
 ]
-

rust/fate_crypto/tests/test_psi_bench.py

@@ -0,0 +1,33 @@
+import random
+import hashlib
+from fate_crypto.psi import Curve25519
+
+
+def ecdh(k, m):
+    return k.encrypt(m)
+
+
+def dh(k, e):
+    return k.diffie_hellman(e)
+
+
+def sha256(value):
+    return hashlib.sha256(bytes(value, encoding="utf-8")).digest()
+
+
+def test_ecdh_encrypt_bench(benchmark):
+    k = Curve25519()
+    m = random.SystemRandom().getrandbits(256).to_bytes(32, "little")
+    result = benchmark(ecdh, k, m)
+
+
+def test_ecdh_dh_bench(benchmark):
+    k = Curve25519()
+    m = random.SystemRandom().getrandbits(256).to_bytes(32, "little")
+    e = k.encrypt(m)
+    result = benchmark(dh, k, e)
+
+
+def test_sha256_bench(benchmark):
+    m = "1000000000"
+    result = benchmark(sha256, m)