Merge pull request #153 from tomato42/test-fixups

Little more test coverage

Author: tomato42

src/ecdsa/test_malformed_sigs.py

@@ -246,7 +246,8 @@ def st_der():
             st.builds(lambda tag, x:
                       encode_constructed(tag, x),
                       st.integers(min_value=0, max_value=0x3f),
-                      st.one_of(children))
+                      st.one_of(children)),
+        max_leaves=40
         )
 
 

src/ecdsa/test_numbertheory.py

@@ -49,6 +49,13 @@ def test_next_prime(prime, next_p):
     assert next_prime(prime) == next_p
 
 
+@pytest.mark.parametrize(
+    "val",
+    [-1, 0, 1])
+def test_next_prime_with_nums_less_2(val):
+    assert next_prime(val) == 2
+
+
 @pytest.mark.parametrize("prime", smallprimes)
 def test_square_root_mod_prime_for_small_primes(prime):
     squares = set()