def test_equality_mat(self, intp_fixture):
""" Equality constraint: abar, bbar, cbar, D
"""
pc, pc_intp = intp_fixture
# number
for i in range(pc_intp.N):
ds = pc_intp.ss[i+1] - pc_intp.ss[i]
ai_new = np.hstack((
pc.abar[i],
pc.abar[i+1] + 2 * ds * pc.bbar[i+1]))
bi_new = np.hstack((pc.bbar[i], pc.bbar[i+1]))
ci_new = np.hstack((pc.cbar[i], pc.cbar[i+1]))
Di_new = block_diag(pc.D[i], pc.D[i+1])
li_new = np.hstack((pc.l[i], pc.l[i+1]))
hi_new = np.hstack((pc.h[i], pc.h[i+1]))
assert np.allclose(ai_new, pc_intp.abar[i])
assert np.allclose(bi_new, pc_intp.bbar[i])
assert np.allclose(ci_new, pc_intp.cbar[i])
assert np.allclose(Di_new, pc_intp.D[i], atol=1e-8)
assert np.allclose(li_new, pc_intp.l[i])
assert np.allclose(hi_new, pc_intp.h[i])
test_interpolate_constraints.py 文件源码
python
阅读 43
收藏 0
点赞 0
评论 0
评论列表
文章目录