summaryrefslogtreecommitdiff
path: root/migrate/tests/fixture
diff options
context:
space:
mode:
Diffstat (limited to 'migrate/tests/fixture')
-rw-r--r--migrate/tests/fixture/__init__.py4
-rw-r--r--migrate/tests/fixture/base.py12
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
"""