Reintegrate test_speed

Related to #139.

Author: kiorky

src/croniter/tests/test_croniter_speed.py

@@ -0,0 +1,121 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, print_function
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
+import os
+import sys
+from datetime import datetime
+from timeit import Timer
+
+import pytz
+
+from croniter import cron_m, croniter
+from croniter.tests import base
+
+
+class CroniterSpeedTest(base.TestCase):
+    def run_long_test(self, iterations=1):
+        dt = datetime(2010, 1, 23, 12, 18)
+        itr = croniter("*/1 * * * *", dt)
+        for i in range(iterations):  # ~ 58
+            itr.get_next()
+
+        itr = croniter("*/5 * * * *", dt)
+        for i in range(iterations):
+            itr.get_next()
+
+        dt = datetime(2010, 1, 24, 12, 2)
+        itr = croniter("0 */3 * * *", dt)
+        for i in range(iterations):
+            itr.get_next()
+
+        dt = datetime(2010, 2, 24, 12, 9)
+        itr = croniter("0 0 */3 * *", dt)
+        for i in range(iterations):
+            itr.get_next(datetime)
+
+        # test leap year
+        dt = datetime(1996, 2, 27)
+        itr = croniter("0 0 * * *", dt)
+        for i in range(iterations):
+            itr.get_next(datetime)
+
+        dt2 = datetime(2000, 2, 27)
+        itr2 = croniter("0 0 * * *", dt2)
+        for i in range(iterations):
+            itr2.get_next(datetime)
+
+        dt = datetime(2010, 2, 25)
+        itr = croniter("0 0 * * sat", dt)
+        for i in range(iterations):
+            itr.get_next(datetime)
+
+        dt = datetime(2010, 1, 25)
+        itr = croniter("0 0 1 * wed", dt)
+        for i in range(iterations):
+            itr.get_next(datetime)
+
+        dt = datetime(2010, 1, 25)
+        itr = croniter("0 0 1 * *", dt)
+        for i in range(iterations):
+            itr.get_next()
+
+        dt = datetime(2010, 8, 25, 15, 56)
+        itr = croniter("*/1 * * * *", dt)
+        for i in range(iterations):
+            itr.get_prev(datetime)
+
+        dt = datetime(2010, 8, 25, 15, 0)
+        itr = croniter("*/1 * * * *", dt)
+        for i in range(iterations):
+            itr.get_prev(datetime)
+
+        dt = datetime(2010, 8, 25, 0, 0)
+        itr = croniter("*/1 * * * *", dt)
+        for i in range(iterations):
+            itr.get_prev(datetime)
+
+        dt = datetime(2010, 8, 25, 15, 56)
+        itr = croniter("0 0 * * sat,sun", dt)
+        for i in range(iterations):
+            itr.get_prev(datetime)
+
+        dt = datetime(2010, 2, 25)
+        itr = croniter("0 0 * * 7", dt)
+        for i in range(iterations):
+            itr.get_prev(datetime)
+
+        # dst regression test
+        tz = pytz.timezone("Europe/Bucharest")
+        offsets = set()
+        dst_cron = "15 0,3 * 3 *"
+        dst_iters = int(2 * 31 * (iterations / 40))
+        dt = datetime(2010, 1, 25, tzinfo=tz)
+        itr = croniter(dst_cron, dt)
+        for i in range(dst_iters):
+            d = itr.get_next(datetime)
+            offsets.add(d.utcoffset())
+        itr = croniter(dst_cron, dt)
+        for i in range(dst_iters):
+            d = itr.get_prev(datetime)
+            offsets.add(d.utcoffset())
+
+    def test_not_long_time(self):
+        if int(sys.version[0]) < 3:
+            return
+        iterations = int(os.environ.get("CRONITER_TEST_SPEED_ITERATIONS", "40"))
+        globs = globals()
+        globs.update(locals())
+        t = Timer("self.run_long_test(iterations)", globals=globs)
+        limit = 80
+        ret = t.timeit(limit)
+        self.assertTrue(ret < limit, "Regression in croniter speed detected ({0} {1}).".format(ret, limit))
+
+
+if __name__ == "__main__":
+    unittest.main()