diff options
Diffstat (limited to 'numpy/polynomial/tests/test_legendre.py')
-rw-r--r-- | numpy/polynomial/tests/test_legendre.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/numpy/polynomial/tests/test_legendre.py b/numpy/polynomial/tests/test_legendre.py index cdfaa96f1..3180db907 100644 --- a/numpy/polynomial/tests/test_legendre.py +++ b/numpy/polynomial/tests/test_legendre.py @@ -423,6 +423,10 @@ class TestFitting(TestCase): # wcoef2d = leg.legfit(x, np.array([yw,yw]).T, 3, w=w) assert_almost_equal(wcoef2d, np.array([coef3,coef3]).T) + # test scaling with complex values x points whose square + # is zero when summed. + x = [1, 1j, -1, -1j] + assert_almost_equal(leg.legfit(x, x, 1), [0, 1]) class TestGauss(TestCase): |