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])