def test_chop_imaginary(self):
array_in = np.array([0.000456789+0.0004j, 1.0456789, 4+0.00004j,
0.0000742+3j, 0.000002, 2+6j])
res = chop(array_in, epsilon=1e-3)
for i, expected in enumerate([0.0+0.0j, 1.0456789, 4+0.0j, 0+3j,
0.0, 2+6j]):
self.assertEqual(expected, res[i])
def test_chop(self):
array_in = [1.023, 1.0456789, 0.0000001, 0.1]
res = chop(array_in, epsilon=1e-3)
for i, expected in enumerate([1.023, 1.0456789, 0.0, 0.1]):
self.assertEqual(expected, res[i])
