14 lines
425 B
Python
14 lines
425 B
Python
import unittest
|
|
|
|
import numpy as np
|
|
|
|
|
|
class TestCase(unittest.TestCase):
|
|
def assertAllClose(self, x1, x2, atol=1e-7, rtol=1e-7):
|
|
np.testing.assert_allclose(x1, x2, atol=atol, rtol=rtol)
|
|
|
|
def assertAlmostEqual(self, x1, x2, decimal=3):
|
|
np.testing.assert_almost_equal(x1, x2, decimal=decimal)
|
|
|
|
def assertLen(self, iterable, expected_len):
|
|
np.testing.assert_equal(len(iterable), expected_len) |