From 7a24608072fb47ee9e40d8127b1f0f552bae0818 Mon Sep 17 00:00:00 2001 From: Joshua Harlow Date: Thu, 30 Oct 2014 13:07:41 -0700 Subject: Fix the digital ocean test on py2.6 The digital ocean datasource test is using assertIs which is only created/existent on py2.7, so for the older py2.6 we need to add similar logic so that the test works correctly there. --- tests/unittests/helpers.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/unittests/helpers.py b/tests/unittests/helpers.py index 9700a4ca..52305397 100644 --- a/tests/unittests/helpers.py +++ b/tests/unittests/helpers.py @@ -35,6 +35,11 @@ else: if PY26: # For now add these on, taken from python 2.7 + slightly adjusted class TestCase(unittest.TestCase): + def assertIs(self, expr1, expr2, msg=None): + if expr1 is not expr2: + standardMsg = '%r is not %r' % (expr1, expr2) + self.fail(self._formatMessage(msg, standardMsg)) + def assertIn(self, member, container, msg=None): if member not in container: standardMsg = '%r not found in %r' % (member, container) -- cgit v1.2.3