forked from p32761584/tensorlayer3
19 lines
382 B
Python
19 lines
382 B
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
from contextlib import contextmanager
|
|
|
|
__all__ = [
|
|
'CustomTestCase',
|
|
]
|
|
|
|
|
|
class CustomTestCase(unittest.TestCase):
|
|
|
|
@contextmanager
|
|
def assertNotRaises(self, exc_type):
|
|
try:
|
|
yield None
|
|
except exc_type:
|
|
raise self.failureException('{} raised'.format(exc_type.__name__))
|