Add testing requirements and update tests to unittest2.

This commit is contained in:
2013-10-27 17:52:50 +00:00
parent a8543521b8
commit 7fbce6c96a
2 changed files with 8 additions and 6 deletions

View File

@@ -1,7 +1,8 @@
import unittest
from .protocol import TS3Proto
import unittest2
from ts3.protocol import TS3Proto
class TS3ProtoTest(unittest.TestCase):
class TS3ProtoTest(unittest2.TestCase):
""" Tests the TS3Proto class """
def setUp(self):
@@ -66,9 +67,9 @@ class TS3ProtoTest(unittest.TestCase):
def suite():
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(TS3ProtoTest))
suite = unittest2.TestSuite()
suite.addTest(unittest2.makeSuite(TS3ProtoTest))
return suite
if __name__ == '__main__':
unittest.main()
unittest2.main()