From 4d09db8b74a95954b980a3ede9fec1801b560c8f Mon Sep 17 00:00:00 2001
From: Ben <ben.curtis@zama.ai>
Date: Thu, 12 May 2022 15:48:30 +0100
Subject: [PATCH] add test code for memory usage

---
 memory_tests/test.py  | 17 +++++++++++++++++
 memory_tests/test2.py | 31 +++++++++++++++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 memory_tests/test.py
 create mode 100644 memory_tests/test2.py

diff --git a/memory_tests/test.py b/memory_tests/test.py
new file mode 100644
index 000000000..22a58d63e
--- /dev/null
+++ b/memory_tests/test.py
@@ -0,0 +1,17 @@
+from estimator_new import *
+from sage.all import oo, save
+
+def test():
+
+    # code
+    D = ND.DiscreteGaussian
+    params = LWE.Parameters(n=1024, q=2 ** 64, Xs=D(0.50, -0.50), Xe=D(2**57), m=oo, tag='TFHE_DEFAULT')
+
+    names = [params, params.updated(n=761), params.updated(q=2 ** 65), params.updated(n=762)]
+
+    for name in names:
+        x = LWE.estimate(name, deny_list=("arora-gb", "bkw"))
+
+    return 0
+
+test()
\ No newline at end of file
diff --git a/memory_tests/test2.py b/memory_tests/test2.py
new file mode 100644
index 000000000..4ad476806
--- /dev/null
+++ b/memory_tests/test2.py
@@ -0,0 +1,31 @@
+
+from multiprocessing import *
+from estimator_new import *
+from sage.all import oo, save
+
+
+def test_memory(x):
+    print("doing job...")
+    print(x)
+    y = LWE.estimate(x, deny_list=("arora-gb", "bkw"))
+    return y
+
+if __name__ == "__main__":
+    D = ND.DiscreteGaussian
+    params = LWE.Parameters(n=1024, q=2 ** 64, Xs=D(0.50, -0.50), Xe=D(2**57), m=oo, tag='TFHE_DEFAULT')
+
+    names = [params, params.updated(n=761), params.updated(q=2**65), params.updated(n=762)]
+    procs = []
+    proc = Process(target=print_func)
+    procs.append(proc)
+    proc.start()
+    p = Pool(1)
+
+    for name in names:
+        proc = Process(target=test_memory, args=(name,))
+        procs.append(proc)
+        proc.start()
+        proc.join()
+
+    for proc in procs:
+        proc.join()
\ No newline at end of file