corrected pspace init file and added test script

komahanb · Feb 3, 2020 · 2be356c · 2be356c
1 parent fd89816
commit 2be356c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/pspace/__init__.py b/pspace/__init__.py
@@ -1,4 +1,3 @@
-from .python import *
 import os
 
 def get_cython_include():

diff --git a/tests/test_pspace.py b/tests/test_pspace.py
@@ -0,0 +1,21 @@
+import pspace.cwrap as uq
+import numpy as np
+
+pfactory = uq.PyParameterFactory()
+y1 = pfactory.createNormalParameter(mu=1.0,sigma=0.1,dmax=2)
+y2 = pfactory.createUniformParameter(a=1.0,b=0.1,dmax=2)
+y3 = pfactory.createExponentialParameter(mu=1.0,beta=0.1,dmax=2)
+
+pc = uq.PyParameterContainer(1)
+pc.addParameter(y1)
+pc.addParameter(y2)
+pc.addParameter(y3)
+
+pc.initialize()
+
+for q in range(pc.getNumQuadraturePoints()):
+    zq, yq = pc.quadrature(q)
+    for k in range(pc.getNumBasisTerms()):
+        print("z=", zq, "q=", q, pc.basis(k, zq))
+
+print(pc.getNumBasisTerms())