Fix broken unit test

mhostetter · Jun 14, 2024 · bfbc148 · bfbc148
1 parent 7809a7d
commit bfbc148
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/sequences/test_preferred_pairs.py b/tests/sequences/test_preferred_pairs.py
@@ -5,22 +5,23 @@
 
 
 def test_exceptions():
+    # NOTE: Need to use next() to get the exception
     with pytest.raises(TypeError):
         # Degree must be an integer
-        sdr.preferred_pairs(6.0)
+        next(sdr.preferred_pairs(6.0))
     with pytest.raises(ValueError):
         # Degree must be positive
-        sdr.preferred_pairs(-1)
+        next(sdr.preferred_pairs(-1))
 
     with pytest.raises(TypeError):
         # Polynomial must be polynomial like
-        sdr.preferred_pairs(6, poly1=1.0)
+        next(sdr.preferred_pairs(6, poly1=1.0))
     with pytest.raises(ValueError):
         # Polynomial must be a primitive polynomial
-        sdr.preferred_pairs(6, poly1=galois.Poly([1, 0, 0, 0, 0, 0, 1]))
+        next(sdr.preferred_pairs(6, poly1=galois.Poly([1, 0, 0, 0, 0, 0, 1])))
     with pytest.raises(ValueError):
         # Polynomial must have correct degree
-        sdr.preferred_pairs(6, poly1=galois.Poly.Degrees([4, 1, 0]))
+        next(sdr.preferred_pairs(6, poly1=galois.Poly.Degrees([4, 1, 0])))
 
 
 def test_degree_divides_4():