diff options
Diffstat (limited to 'migrate/tests/fixture')
-rw-r--r-- | migrate/tests/fixture/__init__.py | 4 | ||||
-rw-r--r-- | migrate/tests/fixture/base.py | 12 |
2 files changed, 5 insertions, 11 deletions
diff --git a/migrate/tests/fixture/__init__.py b/migrate/tests/fixture/__init__.py index 09fc2ca..cfc67b4 100644 --- a/migrate/tests/fixture/__init__.py +++ b/migrate/tests/fixture/__init__.py @@ -1,7 +1,7 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- -import unittest2 +import testtools def main(imports=None): if imports: @@ -10,7 +10,7 @@ def main(imports=None): defaultTest='fixture.suite' else: defaultTest=None - return unittest2.TestProgram(defaultTest=defaultTest) + return testtools.TestProgram(defaultTest=defaultTest) from base import Base from migrate.tests.fixture.pathed import Pathed diff --git a/migrate/tests/fixture/base.py b/migrate/tests/fixture/base.py index 67aabf8..38c91af 100644 --- a/migrate/tests/fixture/base.py +++ b/migrate/tests/fixture/base.py @@ -2,17 +2,11 @@ # -*- coding: utf-8 -*- import re -import unittest2 +import testtools -class Base(unittest2.TestCase): +class Base(testtools.TestCase): - def setup_method(self,func=None): - self.setUp() - - def teardown_method(self,func=None): - self.tearDown() - - def assertEqualsIgnoreWhitespace(self, v1, v2): + def assertEqualIgnoreWhitespace(self, v1, v2): """Compares two strings that should be\ identical except for whitespace """ |