def surf2():
""" Generate surface for testing """
surf = spline.Surface()
surf.degree_u = 2
surf.degree_v = 2
surf.num_ctrlpts_u = 5
surf.num_ctrlpts_v = 5
x = np.arange(0.0, 5.0)
y = np.arange(0.0, 5.0)
ys, xs = np.meshgrid(x, y)
ctrlpt_array = np.column_stack(
(xs.flatten(), ys.flatten(), np.zeros(len(xs.flatten()))))
surf.control_points = ctrlpt_array
uvec = knots.generate_uniform(surf.degree_u, surf.num_ctrlpts_u)
vvec = knots.generate_uniform(surf.degree_v, surf.num_ctrlpts_v)
surf.knot_vector_u = uvec
surf.knot_vector_v = vvec
return surf
def test_surf_control_pt_guard_u():
with pytest.raises(Exception):
x = np.arange(0.0, 5.0)
y = np.arange(0.0, 5.0)
ys, xs = np.meshgrid(x, y)
ctrlpt_array = np.column_stack(
(xs.flatten(), ys.flatten(), np.zeros(len(xs.flatten()))))
surf = spline.Surface()
surf.degree_u = 2
surf.control_points = ctrlpt_array
def test_surf_knot_vector_guard_v():
with pytest.raises(Exception):
x = np.arange(0.0, 5.0)
y = np.arange(0.0, 5.0)
ys, xs = np.meshgrid(x, y)
ctrlpt_array = np.column_stack(
(xs.flatten(), ys.flatten(), np.zeros(len(xs.flatten()))))
surf = spline.Surface()
surf.degree_u = 2
surf.degree_v = 2
num_ctrlpts = 5
knot_vec = knots.generate_uniform(surf.degree_v, num_ctrlpts)
surf.knot_vector_v = knot_vec
kv = knots.generate_uniform(deg, ncpts) cpts = np.linspace(0.0, 4.0, ncpts) ys, xs = np.meshgrid(cpts, cpts) xcpts = xs.flatten() ycpts = ys.flatten() u = np.linspace(0, 1, 800) uu, vv = np.meshgrid(u, u) us = uu.flatten() vs = vv.flatten() uv = np.column_stack((us, vs)) surf = spline.Surface() surf.degree_u = deg surf.degree_v = deg surf.num_ctrlpts_u = ncpts surf.num_ctrlpts_v = ncpts surf.control_points = np.column_stack((xcpts, ycpts, np.zeros(len(xcpts)))) surf.knot_vector_u = kv surf.knot_vector_v = kv foo = surf.points(uv)
