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 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()