import os
from StringIO import StringIO
-import unittest
+from unittest import TestCase
from htsworkflow.util import validate
-class TestValidate(unittest.TestCase):
+class TestValidate(TestCase):
def test_phred33_works(self):
q = StringIO(u"@ abc\nAGCT\n+\nBBBB\n")
errors = validate.validate_fastq(q)
errors = validate.validate_fastq(q)
self.failUnlessEqual(1, errors)
-def suite():
- return unittest.makeSuite(testValidate, 'test')
-if __name__ == "__main__":
- unittest.main(defaultTest='suite')
+def suite():
+ from unittest import TestSuite, defaultTestLoader
+ suite = TestSuite()
+ suite.addTests(defaultTestLoader.loadTestsFromTestCase(testValidate))
+ return suite
+if __name__ == "__main__":
+ from unittest import main
+ main(defaultTest="suite")