| |
| |
|
|
|
|
|
|
|
|
|
|
|
|
| import unittest |
|
|
| import numpy as np |
|
|
| class TestCase(unittest.TestCase): |
|
|
| def setUp(self): |
| np.random.seed(seed=0) |
|
|
| def assertSameOutputs(self, outputs1, outputs2, decimal=7): |
| self.assertEqual(len(outputs1), len(outputs2)) |
| for o1, o2 in zip(outputs1, outputs2): |
| self.assertEqual(o1.dtype, o2.dtype) |
| np.testing.assert_almost_equal(o1, o2, decimal=decimal) |
|
|
| def add_test_case(self, name, test_func): |
| if not name.startswith('test_'): |
| raise ValueError('Test name must start with test_: {}'.format(name)) |
| if hasattr(self, name): |
| raise ValueError('Duplicated test name: {}'.format(name)) |
| setattr(self, name, test_func) |
|
|
