def testLayout(self):
layout = testdata.createSimpleLayout()
(tmpFile, tmpName) = tempfile.mkstemp()
os.close(tmpFile)
# write dataset with layout
p = Point()
p.setName('p1')
p.setLayout(layout)
p2 = Point()
p2.setName('p2')
p2.setLayout(layout)
p['a.1'] = 23
self.assertEqual(p['a.1'], 23)
self.assertRaises(Exception, p.setValue, 'a.4', 34)
ds1 = DataSet()
ds1.setName('ds1')
ds1.addPoint(p)
ds1.addPoint(p2)
ds1.save(tmpName)
# reload dataset
ds2 = DataSet()
ds2.load(tmpName)
self.assertEqual(layout, ds2.layout())
self.assertEqual(ds2.point('p1')['a.1'], 23)
# remove temp file
os.remove(tmpName)
def testLayout(self):
layout = testdata.createSimpleLayout()
(tmpFile, tmpName) = tempfile.mkstemp()
os.close(tmpFile)
# write dataset with layout
p = Point()
p.setName('p1')
p.setLayout(layout)
p2 = Point()
p2.setName('p2')
p2.setLayout(layout)
p['a.1'] = 23
self.assertEqual(p['a.1'], 23)
self.assertRaises(Exception, p.setValue, 'a.4', 34)
ds1 = DataSet()
ds1.setName('ds1')
ds1.addPoint(p)
ds1.addPoint(p2)
ds1.save(tmpName)
# reload dataset
ds2 = DataSet()
ds2.load(tmpName)
self.assertEqual(layout, ds2.layout())
self.assertEqual(ds2.point('p1')['a.1'], 23)
# remove temp file
os.remove(tmpName)
def testHistory(self):
# add a point in a transformed dataset
d = testdata.createSimpleDataSet()
ds = transform(d, 'select', { 'descriptorNames': [ 'a*', 'b*' ] })
p = Point()
p.setName('ptest')
p.setLayout(testdata.createSimpleLayout())
ds.addPoint(p)
def testHistory(self):
# add a point in a transformed dataset
d = testdata.createSimpleDataSet()
ds = transform(d, 'select', {'descriptorNames': ['a*', 'b*']})
p = Point()
p.setName('ptest')
p.setLayout(testdata.createSimpleLayout())
ds.addPoint(p)
def createSequentialDataSet(self):
# used in testRemovePoints
layout = testdata.createSimpleLayout()
ds = DataSet()
p = Point()
p.setLayout(layout)
for name in [ '0', '1', '2', '3', '4', '5' ]:
p.setName(name)
ds.addPoint(p)
return ds
def createSequentialDataSet(self):
# used in testRemovePoints
layout = testdata.createSimpleLayout()
ds = DataSet()
p = Point()
p.setLayout(layout)
for name in ['0', '1', '2', '3', '4', '5']:
p.setName(name)
ds.addPoint(p)
return ds
def testMerge(self):
l2 = testdata.createSimpleLayout()
self.assertEqual(self.l1, l2)
self.assertRaises(Exception, mergeLayouts, self.l1, l2)
l3 = PointLayout()
l3.add('a', RealType)
l3.add('a', '3', RealType)
l4 = mergeLayouts(l2, l3)
self.assertEqual(len(l4.descriptorNames()), len(l2.descriptorNames())+1)
self.assertEqual(l4.descriptorLocation('a').size(RealType, VariableLength), 3)
def testMerge(self):
l2 = testdata.createSimpleLayout()
self.assertEqual(self.l1, l2)
self.assertRaises(Exception, mergeLayouts, self.l1, l2)
l3 = PointLayout()
l3.add('a', RealType)
l3.add('a', '3', RealType)
l4 = mergeLayouts(l2, l3)
self.assertEqual(len(l4.descriptorNames()), len(l2.descriptorNames())+1)
self.assertEqual(l4.descriptorLocation('a').size(RealType, VariableLength), 3)
def testCenter(self):
ds = testdata.createSimpleDataSet()
l = testdata.createSimpleLayout()
for i in range(4):
p = Point()
p.setName('p%d' % i)
p.setLayout(l)
ds.addPoint(p)
ds.removePoint('p')
ds.point('p0')['a.1'] = [ 0, 1 ]
ds.point('p1')['a.1'] = [ 4, 3 ]
ds.point('p2')['a.1'] = [ 6, 9 ]
ds.point('p3')['a.1'] = [ 2, 27 ]
# mean = [ 3, 10 ]
ds = transform(ds, 'fixlength')
dsc = transform(ds, 'center', { 'descriptorNames': 'a.1' })
self.assertEqual(dsc.point('p0')['a.1'], (-3, -9))
self.assertEqual(dsc.point('p1')['a.1'], ( 1, -7))
self.assertEqual(dsc.point('p2')['a.1'], ( 3, -1))
self.assertEqual(dsc.point('p3')['a.1'], (-1, 17))
def testCenter(self):
ds = testdata.createSimpleDataSet()
l = testdata.createSimpleLayout()
for i in range(4):
p = Point()
p.setName("p%d" % i)
p.setLayout(l)
ds.addPoint(p)
ds.removePoint("p")
ds.point("p0")["a.1"] = [0, 1]
ds.point("p1")["a.1"] = [4, 3]
ds.point("p2")["a.1"] = [6, 9]
ds.point("p3")["a.1"] = [2, 27]
# mean = [ 3, 10 ]
ds = transform(ds, "fixlength")
dsc = transform(ds, "center", {"descriptorNames": "a.1"})
self.assertEqual(dsc.point("p0")["a.1"], (-3, -9))
self.assertEqual(dsc.point("p1")["a.1"], (1, -7))
self.assertEqual(dsc.point("p2")["a.1"], (3, -1))
self.assertEqual(dsc.point("p3")["a.1"], (-1, 17))
def setUp(self):
cvar.verbose = False
self.l1 = testdata.createSimpleLayout()
def setUp(self):
cvar.verbose = False
self.l1 = testdata.createSimpleLayout()
