def test_createvectorpolydata_error(): orig = np.random.random((3, 1)) vec = np.random.random((3, 1)) with pytest.raises(ValueError): helpers.vector_poly_data(orig, vec)
def test_createvectorpolydata(): orig = np.random.random((100, 3)) vec = np.random.random((100, 3)) vdata = helpers.vector_poly_data(orig, vec) assert np.any(vdata.points) assert np.any(vdata.point_arrays['vectors'])
def test_createvectorpolydata_1D(): orig = np.random.random(3) vec = np.random.random(3) vdata = helpers.vector_poly_data(orig, vec) assert np.any(vdata.points) assert np.any(vdata.point_data['vectors'])