Source code for woo.tests.io

'''
Test loading and saving woo objects in various formats
'''
from builtins import str
import woo
import unittest
from woo.core import *
from woo.dem import *
from woo.pre import *
from minieigen import *
from woo import utils

[docs]class TestFormatsAndDetection(unittest.TestCase):
[docs] def setUp(self): woo.master.scene=S=Scene(fields=[DemField()]) S.engines=utils.defaultEngines() S.dem.par.add(utils.sphere((0,0,0),radius=1))
[docs] def tryDumpLoad(self,fmt='auto',ext='',load=True): S=woo.master.scene for o in S.engines+[S.dem.par[0]]: out=woo.master.tmpFilename()+ext o.dump(out,format=fmt) if load: o2=Object.load(out,format='auto') self.assertTrue(type(o2)==type(o)) o3=type(o).load(out,format='auto') self.assertRaises(TypeError,lambda: woo.core.Node.load(out))
#if fmt=='expr': print open(out).read()
[docs] def tryDumpLoadStr(self,fmt,load=True): S=woo.master.scene for o in S.engines+[S.dem.par[0]]: dump=o.dumps(format=fmt) if load: Object.loads(dump,format='auto')
[docs] def testRefuseDerivedPyObject(self): 'IO: python-derived objects refuse to save via boost::serialization.' import woo.pre.horse fh=woo.pre.horse.FallingHorse() out=woo.master.tmpFilename()+'.bin.gz' self.assertRaises(IOError,lambda: fh.dump(out)) # this should deted boost::serialization anyway self.assertRaises(IOError,lambda: fh.dump(out,format='boost::serialization'))
[docs] def testStrFile(self): 'IO: file can be given as str' out=woo.master.tmpFilename()+'.expr' woo.master.scene.dem.par[0].dump(out,format='expr')
[docs] def testUnicodeFile(self): 'IO: filename can be given as unicode' out=str(woo.master.tmpFilename()+'.expr') woo.master.scene.dem.par[0].dump(out,format='expr')
[docs] def testExpr(self): 'IO: expression dump/load & format detection (file+string)' self.tryDumpLoad(fmt='expr') self.tryDumpLoadStr(fmt='expr')
[docs] def testJson(self): 'IO: JSON dump/load & format detection (file+string)' self.tryDumpLoad(fmt='json') self.tryDumpLoadStr(fmt='json')
[docs] def testHtml(self): 'IO: HTML dump (file+string)' self.tryDumpLoad(fmt='html',load=False) self.tryDumpLoadStr(fmt='html',load=False)
[docs] def testPickle(self): 'IO: pickle dump/load & format detection (file+string)' self.tryDumpLoad(fmt='pickle') self.tryDumpLoadStr(fmt='pickle',load=True)
[docs] @unittest.skipIf('noxml' in woo.config.features,"Compiled with the 'noxml' feature") def testXml(self): 'IO: XML save/load & format detection' self.tryDumpLoad(ext='.xml')
[docs] @unittest.skipIf('noxml' in woo.config.features,"Compiled with the 'noxml' feature") def testXmlBz2(self): 'IO: XML save/load (bzip2 compressed) & format detection' self.tryDumpLoad(ext='.xml.bz2')
[docs] def testBin(self): 'IO: binary save/load & format detection' self.tryDumpLoad(ext='.bin')
[docs] def testBinGz(self): 'IO: binary save/load (gzip compressed) & format detection' self.tryDumpLoad(ext='.bin.gz')
[docs] def testInvalidFormat(self): 'IO: invalid formats rejected' self.assertRaises(IOError,lambda: woo.master.scene.dem.par[0].dumps(format='bogus'))
[docs] def testInvalidExtension(self): "IO: unrecognized extension rejected with format='auto'" self.assertRaises(IOError,lambda: woo.master.scene.dem.par[0].dump('/tmp/foo.barbaz',format='auto'))
[docs] def testTmpStore(self): 'IO: temporary store (loadTmp, saveTmp)' S=woo.master.scene for o in S.engines+[S.dem.par[0]]: o.saveTmp(quiet=True); o.__class__.loadTmp() # discard the result, but checks type
[docs] def testDeepcopy(self): 'IO: temporary store (Object.deepcopy)' S=woo.master.scene for o in S.engines+[S.dem.par[0]]: o2=o.deepcopy() self.assertTrue(type(o)==type(o2)) self.assertTrue(id(o)!=id(o2))
[docs] def testExprSpecialComments(self): 'IO: special comments #: inside expr dumps' expr=''' #: import os #: g=[] #: for i in range(3): #: g.append((i+1)*os.getpid()) woo.dem.DemField( gravity=g ) ''' field=woo.dem.DemField.loads(expr,format='expr') import os self.assertEqual(field.gravity[0],os.getpid()) self.assertEqual(field.gravity[1],2*os.getpid()) self.assertEqual(field.gravity[2],3*os.getpid())
[docs] def testExprHashColonOverride(self): 'IO: special comments #% inside expr dumps' expr=''' #% a=3 #: b=a woo.dem.DemField(gravity=(a,a,b)) ''' field=woo.core.Object.loads(expr,format='expr',overrideHashPercent={'a':4}) self.assertEqual(field.gravity[0],4) # not 3 as defined in the #: line self.assertRaises(NameError,lambda: woo.core.Object.loads(expr,format='expr',overrideHashPercent={'nonexistent':4}))
[docs]class TestSpecialDumpMethods(unittest.TestCase):
[docs] def setUp(self): woo.master.reset() self.out=woo.master.tmpFilename()
[docs] def testSceneLastDump_direct(self): 'IO: Scene.lastSave set (Object._boostSave overridden)' woo.master.scene.save(self.out) self.assertTrue(woo.master.scene.lastSave==self.out)
[docs]class TestArraySerialization(unittest.TestCase):
[docs] def testMatrixX(self): 'IO: serialization of arrays' sys.stderr.write('A\n') t0=woo.core.WooTestClass() sys.stderr.write('B\n') t0.matX=MatrixX([[0,1,2],[3,4,5]]) sys.stderr.write('C\n') out=woo.master.tmpFilename() t0.save(out) sys.stderr.write('D\n') t1=woo.core.Object.load(out) sys.stderr.write('E\n') self.assertTrue(t1.matX.rows()==2) self.assertTrue(t1.matX.cols()==3) self.assertTrue(t1.matX.sum()==15) sys.stderr.write('F\n')
[docs] def testBoostMultiArray(self): 'IO: serialization of boost::multi_array' t0=woo.core.WooTestClass() t0.arr3d_set((2,2,2),[0,1,2,3,4,5,6,7]) out=woo.master.tmpFilename() t0.save(out) t1=woo.core.Object.load(out) self.assertTrue(t1.arr3d==[[[0.0, 1.0], [2.0, 3.0]], [[4.0, 5.0], [6.0, 7.0]]])