From 202f4544d3738af03c0c45c5076435fca9c08e9a Mon Sep 17 00:00:00 2001 From: Tomas Babej Date: Sun, 22 Mar 2015 16:58:38 +0100 Subject: [PATCH] tests: Python 2.6. does not support timedelta.total_seconds, add workaround --- tasklib/tests.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tasklib/tests.py b/tasklib/tests.py index b740c0d..3e18bc9 100644 --- a/tasklib/tests.py +++ b/tasklib/tests.py @@ -7,6 +7,7 @@ import json import pytz import six import shutil +import sys import tempfile import unittest @@ -36,6 +37,9 @@ TASK_STANDARD_ATTRS = ( 'annotations', ) +total_seconds_2_6 = lambda x: x.microseconds / 1e6 + x.seconds + x.days * 24 * 3600 + + class TasklibTest(unittest.TestCase): def setUp(self): @@ -805,8 +809,12 @@ class DatetimeStringTest(TasklibTest): now = local_zone.localize(datetime.datetime.now()) # Assert that both times are not more than 5 seconds apart - self.assertTrue((now - t['due']).total_seconds() < 5) - self.assertTrue((t['due'] - now).total_seconds() < 5) + if sys.version > (2,6): + self.assertTrue((now - t['due']).total_seconds() < 5) + self.assertTrue((t['due'] - now).total_seconds() < 5) + else: + self.assertTrue(total_seconds_2_6(now - t['due']) < 5) + self.assertTrue(total_seconds_2_6(t['due'] - now) < 5) def test_simple_eoy_conversion(self): if self.tw.version < six.text_type('2.4.0'): -- 2.39.2