From 0918cda254ddb4bc7e69f7dbbb1ff28dbe80fdba Mon Sep 17 00:00:00 2001
From: Danny Willems <be.danny.willems@gmail.com>
Date: Wed, 28 Aug 2024 14:08:19 -0700
Subject: [PATCH] MVPoly: add tests for is_constant

---
 mvpoly/tests/prime.rs | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/mvpoly/tests/prime.rs b/mvpoly/tests/prime.rs
index 8e48a27a48..4294d3ed37 100644
--- a/mvpoly/tests/prime.rs
+++ b/mvpoly/tests/prime.rs
@@ -695,3 +695,24 @@ fn test_degree_random_degree() {
     let degree = unsafe { p.degree() };
     assert!(degree <= max_degree);
 }
+
+#[test]
+fn test_is_constant() {
+    let mut rng = o1_utils::tests::make_test_rng(None);
+    let c = Fp::rand(&mut rng);
+    let p = Dense::<Fp, 4, 5>::from(c);
+    assert!(p.is_constant());
+
+    let p = Dense::<Fp, 4, 5>::zero();
+    assert!(p.is_constant());
+
+    let p = Dense::<Fp, 4, 5>::from_variable(2_usize);
+    assert!(!p.is_constant());
+
+    let p = Dense::<Fp, 4, 5>::from_variable(3_usize);
+    assert!(!p.is_constant());
+
+    // This might be flaky
+    let p = unsafe { Dense::<Fp, 4, 5>::random(&mut rng, None) };
+    assert!(!p.is_constant());
+}