import os
-from unittest2 import TestCase
+from unittest import TestCase
try:
from xml.etree import ElementTree
self.failUnless(flatten(self.foo_tree), 'asdf')
def suite():
- from unittest2 import TestSuite, defaultTestLoader
+ from unittest import TestSuite, defaultTestLoader
suite = TestSuite()
suite.addTests(defaultTestLoader.loadTestsFromTestCase(testETHelper))
return suite
if __name__ == "__main__":
- from unittest2 import main
+ from unittest import main
main(defaultTest="suite")