def testFromDomNumarrayVector(self):
p3 = ParameterSet()
mynumpy = numpy.array([3., 4., 5.], dtype=numpy.float64)
p3.declareParameter(numtest=mynumpy)
doc = self._class(p3)
assert doc.numtest.dtype == numpy.float64
assert type(doc.numtest) == numpy.ndarray
def testParameterSetWithChildrenFromXML(self):
p2 = ParameterSet()
p2.declareParameter(s="abc", f=3.)
self.p.declareParameter(child=p2)
doc = self._class(self.p)
#pset = ParameterSet.fromDom(doc.getElementsByTagName("ParameterSet")[0])
self.assertEqual(self.p.child.f, doc.child.f)
def testDatasource(self):
p5 = ParameterSet()
myURI = DataSource("somelocalfile.txt", "text")
p5.declareParameter(uritest=myURI)
doc = self._class(p5)
self.assertEqual(myURI.uri, doc.uritest.uri)
self.assertEqual(myURI.fileformat, doc.uritest.fileformat)
assert type(doc.uritest) == DataSource
def testStringLists(self):
p4 = ParameterSet()
mylist = ["a", "b", "c"]
p4.declareParameter(listest=mylist)
doc = self._class(p4)
assert type(doc.listest) == list
assert len(doc.listest) == len(mylist)
assert min(
[mylist[i] == doc.listest[i] for i in range(len(doc.listest))])
def testFloatLists(self):
p4 = ParameterSet()
mylist = [1., 2., 4.]
p4.declareParameter(listest=mylist)
doc = self._class(p4)
assert type(doc.listest) == numpy.ndarray
assert doc.listest.dtype == numpy.float_
assert len(doc.listest) == len(mylist)
assert min(
[mylist[i] == doc.listest[i] for i in range(len(doc.listest))])
def testBoolLists(self):
p4 = ParameterSet()
mylist = [True, False, False, True]
p4.declareParameter(listest=mylist)
doc = self._class(p4)
assert type(doc.listest) == numpy.ndarray
assert doc.listest.dtype == numpy.bool_
assert len(doc.listest) == len(mylist)
assert min(
[mylist[i] == doc.listest[i] for i in range(len(doc.listest))])
def testFromDomInt(self):
p3 = ParameterSet()
p3.declareParameter(inttest=1)
doc = self._class(p3)
assert type(doc.inttest) == int
def testParameterSetChild(self):
p2 = ParameterSet()
p2.declareParameter(s="abc", f=3.)
self.p.declareParameter(child=p2)
self.assertEqual(self.p.child.s, "abc")
self.assertEqual(self.p.child.f, 3.)
def setUp(self):
self.p = ParameterSet()
self.p.declareParameter(gamma1=1., gamma2=2., gamma3=3.)