| import unittest |
| import numpy as np |
|
|
|
|
| |
| def sigfigs(n, sf): |
| float('%.1g' % n) |
|
|
|
|
| class ExtendedTest(unittest.TestCase): |
| def assertAlmostEqual(self, a, b, places=7, msg='', delta=None, sigfigs=None): |
| if sigfigs is None: |
| super(ExtendedTest, self).assertAlmostEqual(a, b, places, msg, delta) |
| else: |
| a_ = float(('%%.%dg' % sigfigs) % a) |
| b_ = float(('%%.%dg' % sigfigs) % b) |
| if a_ != b_: |
| raise AssertionError(msg or "%f != %f to %d significant figures (%f != %f)" % (a, b, sigfigs, a_, b_)) |
|
|
| def assertDictOfArraysEqual(self, a, b, msg=''): |
| self.assertIsInstance(a, dict, msg or 'First argument is not a dictionary') |
| self.assertIsInstance(b, dict, msg or 'Second argument is not a dictionary') |
| self.assertSetEqual(set(a.keys()), set(b.keys()), msg or 'Keys do not match') |
| for k in a.keys(): |
| if isinstance(a[k], np.ndarray) and isinstance(b[k], np.ndarray): |
| np.testing.assert_array_equal(a[k], b[k], err_msg=msg + "\nwith key [%s]" % (k)) |
| else: |
| np.testing.assert_array_equal(np.array(a[k]), np.array(b[k]), err_msg=msg + "\nwith key [%s]" % (k)) |
|
|
| def assertDictOfArraysAlmostEqual(self, a, b, decimal=6, msg=''): |
| self.assertIsInstance(a, dict, msg or 'First argument is not a dictionary') |
| self.assertIsInstance(b, dict, msg or 'Second argument is not a dictionary') |
| self.assertSetEqual(set(a.keys()), set(b.keys()), msg or 'Keys do not match') |
| for k in a.keys(): |
| if isinstance(a[k], np.ndarray) and isinstance(b[k], np.ndarray): |
| np.testing.assert_array_almost_equal(a[k], b[k], decimal=decimal, err_msg=msg + "\nwith key [%s]" % (k)) |
| else: |
| np.testing.assert_array_almost_equal(np.array(a[k]), np.array(b[k]), decimal=decimal, err_msg=msg + "\nwith key [%s]" % (k)) |
|
|
