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 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 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 testParameterSetFromXML(self):
s = StringIO()
self.p.writeXML(s)
s.seek(0)
xmlout = s.read()
esysxml = ESySXMLParser(xmlout)
doc = self._class(self.p)
pset = ParameterSet.fromDom(
esysxml,
self._dom(self.p).getElementsByTagName("ParameterSet")[0])
assert (isinstance(pset, ParameterSet))
assert (isinstance(doc, ParameterSet))
self.assertEqual(self.p.gamma1, doc.gamma1)
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.)
class Test_ParameterSetTestCase(unittest.TestCase):
def setUp(self):
self.p = ParameterSet()
self.p.declareParameter(gamma1=1., gamma2=2., gamma3=3.)
def testParameterSetCreation(self):
self.assertEqual(self.p.gamma1, 1.)
def testParameterSetXMLCreation(self):
s = StringIO()
self.p.writeXML(s)
s.seek(0)
xmlout = s.read()
assert ("gamma1" in xmlout)
assert ("gamma2" in xmlout)
assert ("gamma3" in xmlout)
esysxml = ESySXMLParser(xmlout)
parsable = esysxml.parse()[0]
assert (isinstance(parsable, ParameterSet))
assert (self._dom(self.p).getElementsByTagName("ParameterSet"))
def testParameterSetFromXML(self):
s = StringIO()
self.p.writeXML(s)
s.seek(0)
xmlout = s.read()
esysxml = ESySXMLParser(xmlout)
doc = self._class(self.p)
pset = ParameterSet.fromDom(
esysxml,
self._dom(self.p).getElementsByTagName("ParameterSet")[0])
assert (isinstance(pset, ParameterSet))
assert (isinstance(doc, ParameterSet))
self.assertEqual(self.p.gamma1, doc.gamma1)
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 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 _dom(self, input):
s = StringIO()
input.writeXML(s)
s.seek(0)
xmlout = s.read()
doc = minidom.parseString(xmlout)
return doc
def _class(self, input):
s = StringIO()
input.writeXML(s)
s.seek(0)
xmlout = s.read()
esysxml = ESySXMLParser(xmlout)
doc = esysxml.parse()[0]
return doc
def testFromDomInt(self):
p3 = ParameterSet()
p3.declareParameter(inttest=1)
doc = self._class(p3)
assert type(doc.inttest) == int
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 testFromDomNumarrayMulti(self):
p3 = ParameterSet()
mynumpy = numpy.array([[1., 2., 3.], [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 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 testIntLists(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.int_
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 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 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