Add more tests for NumberRange

sqlalchemy_utils/primitives/__init__.py

@@ -1,4 +1,6 @@
-from .number_range import NumberRange, NumberRangeException
+from .number_range import (
+    NumberRange, NumberRangeException, RangeBoundsException
+)
 from .weekday import WeekDay
 from .weekdays import WeekDays
 
@@ -6,6 +8,7 @@ from .weekdays import WeekDays
 __all__ = (
     NumberRange,
     NumberRangeException,
+    RangeBoundsException,
     WeekDay,
     WeekDays
 )

tests/primitives/test_number_range.py

@@ -1,5 +1,7 @@
-from pytest import raises
+from pytest import raises, mark
-from sqlalchemy_utils.primitives import NumberRange, NumberRangeException
+from sqlalchemy_utils.primitives import (
+    NumberRange, NumberRangeException, RangeBoundsException
+)
 
 
 class TestNumberRangeInit(object):
@@ -76,9 +78,19 @@ def test_str_representation():
     assert str(NumberRange(1, 1)) == '1'
 
 
-def test_raises_exception_for_badly_constructed_range():
+
-    with raises(NumberRangeException):
+@mark.parametrize('number_range',
-        NumberRange(3, 2)
+    (
+        (3, 2),
+        [4, 2],
+        '5-2',
+        (float('inf'), 2),
+        '[4, 3]',
+    )
+)
+def test_raises_exception_for_badly_constructed_range(number_range):
+    with raises(RangeBoundsException):
+        NumberRange(number_range)
 
 
 class TestArithmeticOperators(object):