Increase NTT code coverage

mhostetter · May 12, 2022 · c5d930d · c5d930d
1 parent eadc6a3
commit c5d930d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/fields/test_ntt.py b/tests/fields/test_ntt.py
@@ -24,6 +24,8 @@ def test_ntt_exceptions():
  galois.ntt([1, 2, 3, 4], size=6.0)
  with pytest.raises(TypeError):
  galois.ntt([1, 2, 3, 4], modulus=3*256 + 1.0)
+ with pytest.raises(TypeError):
+ galois.ntt([1, 2, 3, 4], forward=1.0)
 
  with pytest.raises(ValueError):
  GF = galois.GF(2**8) # Invalid field for NTTs
@@ -32,6 +34,10 @@ def test_ntt_exceptions():
  galois.ntt([1, 2, 3, 4], size=3)
  with pytest.raises(ValueError):
  galois.ntt([1, 2, 3, 40], modulus=13)
+ with pytest.raises(ValueError):
+ galois.ntt([1, 2, 3, 40], modulus=43)
+ with pytest.raises(ValueError):
+ galois.ntt([1, 2, 3, 4], modulus=3*256 + 2)
 
 
 @pytest.mark.parametrize(["x", "p", "X"], NTT_LUTS)