def test_03(self):
x = np.arange(10).astype(dtype=np.float).reshape((1, -1))
y = np.arange(10).astype(dtype=np.float)
p = _ndpolyfit(x, y)
np.testing.assert_almost_equal(p, [1.0, 0.0])
def test_16(self):
x = np.arange(10).astype(dtype=np.float)
y = np.concatenate((np.arange(10).astype(dtype=np.float).reshape(
(-1, 1)), np.arange(10).astype(dtype=np.float).reshape((-1, 1))),
axis=1)
p = _ndpolyfit(x, y, missing_value=5)
np.testing.assert_almost_equal(p, [[1.0, 1.0], [0.0, 0.0]])
def test_10(self):
x = np.arange(10).astype(dtype=np.float)
y = np.arange(10).astype(dtype=np.float)
y[5] = np.nan
p = _ndpolyfit(x, y)
np.testing.assert_almost_equal(p, [1.0, 0.0])
def test_14(self):
x = np.arange(10).astype(dtype=np.float)
y = np.concatenate((np.arange(10).astype(dtype=np.float).reshape(
(-1, 1)), np.arange(10).astype(dtype=np.float).reshape((-1, 1))),
axis=1)
y[4, :] = np.nan
y[2, :] = np.nan
p = _ndpolyfit(x, y)
np.testing.assert_almost_equal(p, [[1.0, 1.0], [0.0, 0.0]])
def test_11(self):
x = np.arange(10).astype(dtype=np.float)
y = np.arange(10).astype(dtype=np.float)
for i in range(20):
idx = np.random.randint(0, 10)
y[idx] = np.nan
p = _ndpolyfit(x, y)
y[idx] = idx
np.testing.assert_almost_equal(p, [1.0, 0.0])
def test_06(self):
x = np.arange(-10, 10).astype(dtype=np.float)
y = np.arange(-10, 10).astype(dtype=np.float)
y = np.moveaxis(np.tile(y, (4, 3, 1)), 2, 0)
actual_p = _ndpolyfit(x, y)
expected_p = np.moveaxis(np.tile(np.asarray([1.0, 0.0]), (4, 3, 1)), 2,
0)
np.testing.assert_almost_equal(actual_p, expected_p)
def test_05(self):
x = np.arange(10).astype(dtype=np.float)
for i in range(50):
expected_p = np.random.randint(-10, 10, size=2)
while expected_p[0] == 0:
expected_p = np.random.randint(-10, 10, size=2)
y = expected_p[0] * x + expected_p[1]
actual_p = _ndpolyfit(x, y)
np.testing.assert_almost_equal(expected_p, actual_p)
def test_09(self):
x = np.arange(-10, 10).astype(dtype=np.float)
max_dim = 6
max_dim_size = 11
for i in range(50):
expected_p = np.random.randint(-10, 10, size=2)
while expected_p[0] == 0:
expected_p = np.random.randint(-10, 10, size=2)
y = expected_p[0] * x + expected_p[1]
other_dims = np.random.randint(1, max_dim_size,
np.random.randint(1, max_dim))
axis = np.random.randint(0, other_dims.ndim + 1)
y = np.moveaxis(np.tile(y, (*other_dims, 1)), -1, axis)
expected_p = np.moveaxis(np.tile(expected_p, (*other_dims, 1)), -1,
axis)
actual_p = _ndpolyfit(x, y, axis=axis)
np.testing.assert_almost_equal(expected_p, actual_p)
def test_04(self):
x = np.arange(10).astype(dtype=np.float).reshape((2, 5))
y = np.arange(10).astype(dtype=np.float)
with self.assertRaises(ValueError):
p = _ndpolyfit(x, y)
