From 209043e54c7bce5ec40231314d356950db9f03eb Mon Sep 17 00:00:00 2001 From: Konsta Vesterinen Date: Tue, 14 Jan 2014 10:11:13 +0200 Subject: [PATCH] More tests for comparator --- tests/types/test_int_range.py | 34 +++++++++++++++++++++++++++++++++- 1 file changed, 33 insertions(+), 1 deletion(-) diff --git a/tests/types/test_int_range.py b/tests/types/test_int_range.py index c60135c..65397e1 100644 --- a/tests/types/test_int_range.py +++ b/tests/types/test_int_range.py @@ -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