test/python/run_contour.py

from esys.pyvisi import DataCollector, Scene, Contour
from esys.pyvisi.constant import *
import unittest, os
from stat import ST_SIZE

try:
     PYVISI_WORKDIR=os.environ['PYVISI_WORKDIR']
except KeyError:
     PYVISI_WORKDIR='.'

try:
     PYVISI_TEST_DATA_ROOT=os.environ['PYVISI_TEST_DATA_ROOT']
except KeyError:
     PYVISI_TEST_DATA_ROOT='.'
 

PYVISI_TEST_MESHES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT,"data_meshes")
PYVISI_TEST_CONTOUR_IMAGES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT, "data_sample_images","contour")
MIN_IMAGE_SIZE = 100
FILE_2D = "interior_2D.xml"

X_SIZE = 400
Y_SIZE = 400

JPG_RENDERER = Renderer.OFFLINE_JPG

class TestContour(unittest.TestCase):
    def setUp(self):
        self.scene = \
                Scene(renderer = JPG_RENDERER, num_viewport = 1,
                        x_size = X_SIZE, y_size = Y_SIZE)

        self.data_collector = DataCollector(source = Source.XML)
        self.data_collector.setFileName(file_name = \
                os.path.join(PYVISI_TEST_MESHES_PATH,FILE_2D))

        self.contour = Contour(scene = self.scene,
                data_collector = self.data_collector,
                viewport = Viewport.SOUTH_WEST, lut = Lut.COLOR,
                cell_to_point = False, outline = True)

    def tearDown(self):
        self.scene
        self.data_collector
        self.contour

    def render(self, file):
        self.scene.render(image_name = \
        os.path.join(PYVISI_TEST_CONTOUR_IMAGES_PATH ,file))

        self.failUnless(os.stat(os.path.join(PYVISI_TEST_CONTOUR_IMAGES_PATH ,file))[ST_SIZE] > MIN_IMAGE_SIZE)

    def testGenerateContours(self):
        self.contour.generateContours(5)
        self.render("TestContour_testGenerateContours.jpg")


##############################################################################


if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(unittest.TestLoader().loadTestsFromTestCase(TestContour))
    unittest.TextTestRunner(verbosity=2).run(suite)

1	from esys.pyvisi import DataCollector, Scene, Contour
2	from esys.pyvisi.constant import *
3	import unittest, os
4	from stat import ST_SIZE
5
6	try:
7	PYVISI_WORKDIR=os.environ['PYVISI_WORKDIR']
8	except KeyError:
9	PYVISI_WORKDIR='.'
10
11	try:
12	PYVISI_TEST_DATA_ROOT=os.environ['PYVISI_TEST_DATA_ROOT']
13	except KeyError:
14	PYVISI_TEST_DATA_ROOT='.'
15
16
17
18	PYVISI_TEST_MESHES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT,"data_meshes")
19	PYVISI_TEST_CONTOUR_IMAGES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT, "data_sample_images","contour")
20	MIN_IMAGE_SIZE = 100
21	FILE_2D = "interior_2D.xml"
22
23	X_SIZE = 400
24	Y_SIZE = 400
25
26	JPG_RENDERER = Renderer.OFFLINE_JPG
27
28	class TestContour(unittest.TestCase):
29	def setUp(self):
30	self.scene = \
31	Scene(renderer = JPG_RENDERER, num_viewport = 1,
32	x_size = X_SIZE, y_size = Y_SIZE)
33
34	self.data_collector = DataCollector(source = Source.XML)
35	self.data_collector.setFileName(file_name = \
36	os.path.join(PYVISI_TEST_MESHES_PATH,FILE_2D))
37
38	self.contour = Contour(scene = self.scene,
39	data_collector = self.data_collector,
40	viewport = Viewport.SOUTH_WEST, lut = Lut.COLOR,
41	cell_to_point = False, outline = True)
42
43	def tearDown(self):
44	self.scene
45	self.data_collector
46	self.contour
47
48	def render(self, file):
49	self.scene.render(image_name = \
50	os.path.join(PYVISI_TEST_CONTOUR_IMAGES_PATH ,file))
51
52	self.failUnless(os.stat(os.path.join(PYVISI_TEST_CONTOUR_IMAGES_PATH ,file))[ST_SIZE] > MIN_IMAGE_SIZE)
53
54	def testGenerateContours(self):
55	self.contour.generateContours(5)
56	self.render("TestContour_testGenerateContours.jpg")
57
58
59	##############################################################################
60
61
62	if __name__ == '__main__':
63	suite = unittest.TestSuite()
64	suite.addTest(unittest.TestLoader().loadTestsFromTestCase(TestContour))
65	unittest.TextTestRunner(verbosity=2).run(suite)
66