From f764ea5cf5c3bba742e975837aa69e4d24752b9e Mon Sep 17 00:00:00 2001
From: nindanaoto <matsuoka.kotaro@gmail.com>
Date: Tue, 20 Feb 2024 16:06:50 +0000
Subject: [PATCH] Added lvl0test

---
 test/gate.cpp | 92 +++++++++++++++++++++++++++++++++++----------------
 1 file changed, 63 insertions(+), 29 deletions(-)

diff --git a/test/gate.cpp b/test/gate.cpp
index 3e2f123..794fd1f 100644
--- a/test/gate.cpp
+++ b/test/gate.cpp
@@ -144,39 +144,73 @@ void RunTest()
     vector<TLWE<P>> cres(kNumTests);
     vector<TLWE<P>> c(3 * kNumTests);
 
-    Test<P>("NOT", TFHEpp::HomNOT<P>, NotChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("COPY", TFHEpp::HomCOPY<P>, CopyChegk, p, cres, c, kNumTests, *sk,
-            ek);
-    Test<P>("NAND", TFHEpp::HomNAND<lvl10param, lvl01param, lvl1param::μ>,
-            NandChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("OR", TFHEpp::HomOR<lvl10param, lvl01param, lvl1param::μ>, OrChegk,
-            p, cres, c, kNumTests, *sk, ek);
-    Test<P>("ORYN", TFHEpp::HomORYN<lvl10param, lvl01param, lvl1param::μ>,
-            OrYNChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("ORNY", TFHEpp::HomORNY<lvl10param, lvl01param, lvl1param::μ>,
-            OrNYChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("AND", TFHEpp::HomAND<lvl10param, lvl01param, lvl1param::μ>,
-            AndChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("ANDYN", TFHEpp::HomANDYN<lvl10param, lvl01param, lvl1param::μ>,
-            AndYNChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("ANDNY", TFHEpp::HomANDNY<lvl10param, lvl01param, lvl1param::μ>,
-            AndNYChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("XOR", TFHEpp::HomXOR<lvl10param, lvl01param, lvl1param::μ>,
-            XorChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("XNOR", TFHEpp::HomXNOR<lvl10param, lvl01param, lvl1param::μ>,
-            XnorChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("MUX", TFHEpp::HomMUX<P>, MuxChegk, p, cres, c, kNumTests, *sk, ek);
-    Test<P>("NMUX", TFHEpp::HomNMUX<P>, NMuxChegk, p, cres, c, kNumTests, *sk,
-            ek);
-    Test<P>("ConstantZero", TFHEpp::HomCONSTANTZERO<P>, ConstantZeroChegk, p,
-            cres, c, kNumTests, *sk, ek);
-    Test<P>("ConstantOne", TFHEpp::HomCONSTANTONE<P>, ConstantOneChegk, p, cres,
-            c, kNumTests, *sk, ek);
+    if constexpr (std::is_same_v<P, lvl0param>) {
+        cout << "lvl0param" << endl;
+                Test<P>("NOT", TFHEpp::HomNOT<P>, NotChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("COPY", TFHEpp::HomCOPY<P>, CopyChegk, p, cres, c, kNumTests, *sk,
+                        ek);
+                Test<P>("NAND", TFHEpp::HomNAND<lvl01param, lvl1param::μ, lvl10param>,
+                        NandChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("OR", TFHEpp::HomOR<lvl01param, lvl1param::μ, lvl10param>, OrChegk,
+                        p, cres, c, kNumTests, *sk, ek);
+                Test<P>("ORYN", TFHEpp::HomORYN<lvl01param, lvl1param::μ, lvl10param>,
+                        OrYNChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("ORNY", TFHEpp::HomORNY<lvl01param, lvl1param::μ, lvl10param>,
+                        OrNYChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("AND", TFHEpp::HomAND<lvl01param, lvl1param::μ, lvl10param>,
+                        AndChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("ANDYN", TFHEpp::HomANDYN<lvl01param, lvl1param::μ, lvl10param>,
+                        AndYNChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("ANDNY", TFHEpp::HomANDNY<lvl01param, lvl1param::μ, lvl10param>,
+                        AndNYChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("XOR", TFHEpp::HomXOR<lvl01param, lvl1param::μ, lvl10param>,
+                        XorChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("XNOR", TFHEpp::HomXNOR<lvl01param, lvl1param::μ, lvl10param>,
+                        XnorChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("MUX", TFHEpp::HomMUX<P>, MuxChegk, p, cres, c, kNumTests, *sk, ek);
+                Test<P>("NMUX", TFHEpp::HomNMUX<P>, NMuxChegk, p, cres, c, kNumTests, *sk,
+                        ek);
+                Test<P>("ConstantZero", TFHEpp::HomCONSTANTZERO<P>, ConstantZeroChegk, p,
+                        cres, c, kNumTests, *sk, ek);
+                Test<P>("ConstantOne", TFHEpp::HomCONSTANTONE<P>, ConstantOneChegk, p, cres,
+                        c, kNumTests, *sk, ek);
+    }
+    else if constexpr (std::is_same_v<P, lvl1param>) {
+        cout << "lvl1param" << endl;
+        Test<P>("NOT", TFHEpp::HomNOT<P>, NotChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("COPY", TFHEpp::HomCOPY<P>, CopyChegk, p, cres, c, kNumTests, *sk,
+                ek);
+        Test<P>("NAND", TFHEpp::HomNAND<lvl10param, lvl01param, lvl1param::μ>,
+                NandChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("OR", TFHEpp::HomOR<lvl10param, lvl01param, lvl1param::μ>, OrChegk,
+                p, cres, c, kNumTests, *sk, ek);
+        Test<P>("ORYN", TFHEpp::HomORYN<lvl10param, lvl01param, lvl1param::μ>,
+                OrYNChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("ORNY", TFHEpp::HomORNY<lvl10param, lvl01param, lvl1param::μ>,
+                OrNYChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("AND", TFHEpp::HomAND<lvl10param, lvl01param, lvl1param::μ>,
+                AndChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("ANDYN", TFHEpp::HomANDYN<lvl10param, lvl01param, lvl1param::μ>,
+                AndYNChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("ANDNY", TFHEpp::HomANDNY<lvl10param, lvl01param, lvl1param::μ>,
+                AndNYChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("XOR", TFHEpp::HomXOR<lvl10param, lvl01param, lvl1param::μ>,
+                XorChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("XNOR", TFHEpp::HomXNOR<lvl10param, lvl01param, lvl1param::μ>,
+                XnorChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("MUX", TFHEpp::HomMUX<P>, MuxChegk, p, cres, c, kNumTests, *sk, ek);
+        Test<P>("NMUX", TFHEpp::HomNMUX<P>, NMuxChegk, p, cres, c, kNumTests, *sk,
+                ek);
+        Test<P>("ConstantZero", TFHEpp::HomCONSTANTZERO<P>, ConstantZeroChegk, p,
+                cres, c, kNumTests, *sk, ek);
+        Test<P>("ConstantOne", TFHEpp::HomCONSTANTONE<P>, ConstantOneChegk, p, cres,
+                c, kNumTests, *sk, ek);
+    }
 }
 
 int main()
 {
     RunTest<lvl1param>();
-    // RunTest<lvl0param>();
+    RunTest<lvl0param>();
     return 0;
 }
\ No newline at end of file