More tests for comparator
This commit is contained in:
@@ -79,7 +79,6 @@ class NumberRangeTestCase(TestCase):
|
||||
assert building.persons_at_night.upper == 15
|
||||
|
||||
|
||||
|
||||
class TestIntRangeTypeOnPostgres(NumberRangeTestCase):
|
||||
dns = 'postgres://postgres@localhost/sqlalchemy_utils_test'
|
||||
|
||||
@@ -134,5 +133,38 @@ class TestIntRangeTypeOnPostgres(NumberRangeTestCase):
|
||||
)
|
||||
assert query.count()
|
||||
|
||||
@mark.parametrize(
|
||||
'number_range',
|
||||
(
|
||||
[1, 4],
|
||||
4,
|
||||
[2, inf]
|
||||
)
|
||||
)
|
||||
def test_le_operator(self, number_range):
|
||||
self.create_building([1, 3])
|
||||
query = (
|
||||
self.session.query(self.Building)
|
||||
.filter(self.Building.persons_at_night <= number_range)
|
||||
)
|
||||
assert query.count()
|
||||
|
||||
@mark.parametrize(
|
||||
'number_range',
|
||||
(
|
||||
[2, 4],
|
||||
4,
|
||||
[1, inf]
|
||||
)
|
||||
)
|
||||
def test_lt_operator(self, number_range):
|
||||
self.create_building([1, 3])
|
||||
query = (
|
||||
self.session.query(self.Building)
|
||||
.filter(self.Building.persons_at_night < number_range)
|
||||
)
|
||||
assert query.count()
|
||||
|
||||
|
||||
class TestNumberRangeTypeOnSqlite(NumberRangeTestCase):
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user