Source code for woo.tests.batch

# encoding: utf-8
# 2013 © Václav Šmilauer <eu@doxos.eu>

import unittest
from minieigen import *
import woo.dem, woo.core, woo.pre, woo.batch
import sys, math, numpy

PY3K=(sys.version_info[0]==3)

class TestBatchResults(unittest.TestCase):
    def setUp(self):
        self.misc=dict(a='a',pi=math.pi,sphere=woo.dem.Sphere(radius=2.3))
        self.scene=woo.core.Scene(fields=[woo.dem.DemField()])
        self.scene.pre=woo.pre.horse.FallingHorse()
        self.series=dict(random=numpy.random.rand(100),ones=numpy.ones(10))
        self.series['group1/item1']=numpy.ones(10)
        self.series['group1/item2']=2*numpy.ones(10)
        self.series['group2/item1']=3*numpy.ones(10)
        self.numE=len(self.scene.engines)
    def _checkDbContents(self,db):
        r=woo.batch.dbReadResults(db)[0]
        self.assertTrue(isinstance(r['pre'],woo.pre.horse.FallingHorse))
        self.assertTrue(r['misc']['a']=='a')
        self.assertTrue(r['misc']['pi']==math.pi)
        self.assertTrue(r['misc']['sphere'].radius==2.3)
        self.assertTrue(r['series']['ones'][2]==1)
        self.assertEqual(len(r['engines']),self.numE)
    def _writeDb(self,db):
        woo.batch.writeResults(self.scene,defaultDb=db,syncXls=False,quiet=True,series=self.series,**self.misc)
    def _writeXls(self,db,xls):
        woo.batch.dbToSpread(db,xls)
    def _writeCsv(self,db,out):
        woo.batch.dbToSpread(db,out,series=False)
    #def testSqlite(self):
    #    'Batch: results in SQLite (store/read/XLS(X)/CSV)'
    #    db=woo.master.tmpFilename()+'.sqlite'
    #    self._writeDb(db)
    #    self._checkDbContents(db)
    #    self._writeXls(db,db+'.xlsx')
    #    self._writeCsv(db,db+'.csv')
    @unittest.skipIf(sys.platform=='win32','HDF5 not supported under Windows yet.')    
    def testHdf5(self):
        'Batch: results in HDF5 (store/read/XLS(X)/CSV)'
        db=woo.master.tmpFilename()+'.hdf5'
        self._writeDb(db)
        self._checkDbContents(db)
        self._writeXls(db,db+'.xlsx')
        self._writeCsv(db,db+'.csv')