add test for rate limit comparison (lt)

alisaifee · Oct 5, 2015 · 72d1692 · 72d1692
1 parent 14b9739
commit 72d1692
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_limit_granularities.py b/tests/test_limit_granularities.py
@@ -18,3 +18,10 @@ def test_representation(self):
         self.assertTrue("1 per 1 day" in str(limits.RateLimitItemPerDay(1)))
         self.assertTrue("1 per 1 month" in str(limits.RateLimitItemPerMonth(1)))
         self.assertTrue("1 per 1 year" in str(limits.RateLimitItemPerYear(1)))
+
+    def test_comparison(self):
+        self.assertTrue(limits.RateLimitItemPerSecond(1) < limits.RateLimitItemPerMinute(1))
+        self.assertTrue(limits.RateLimitItemPerMinute(1) < limits.RateLimitItemPerHour(1))
+        self.assertTrue(limits.RateLimitItemPerHour(1) < limits.RateLimitItemPerDay(1))
+        self.assertTrue(limits.RateLimitItemPerDay(1) < limits.RateLimitItemPerMonth(1))
+        self.assertTrue(limits.RateLimitItemPerMonth(1) < limits.RateLimitItemPerYear(1))