test_regression.py 文件源码

def test_polydiv_type(self):
        # Make polydiv work for complex types
        msg = "Wrong type, should be complex"
        x = np.ones(3, dtype=np.complex)
        q, r = np.polydiv(x, x)
        assert_(q.dtype == np.complex, msg)
        msg = "Wrong type, should be float"
        x = np.ones(3, dtype=np.int)
        q, r = np.polydiv(x, x)
        assert_(q.dtype == np.float, msg)