3 |
import unittest, os |
import unittest, os |
4 |
from stat import ST_SIZE |
from stat import ST_SIZE |
5 |
|
|
6 |
PYVISI_TEST_MESHES_PATH = "data_meshes/" |
try: |
7 |
PYVISI_TEST_CARPET_IMAGES_PATH = "data_sample_images/carpet/" |
PYVISI_WORKDIR=os.environ['PYVISI_WORKDIR'] |
8 |
|
except KeyError: |
9 |
|
PYVISI_WORKDIR='.' |
10 |
|
try: |
11 |
|
PYVISI_TEST_DATA_ROOT=os.environ['PYVISI_TEST_DATA_ROOT'] |
12 |
|
except KeyError: |
13 |
|
PYVISI_TEST_DATA_ROOT='.' |
14 |
|
|
15 |
|
PYVISI_TEST_MESHES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT,"data_meshes") |
16 |
|
PYVISI_TEST_CARPET_IMAGES_PATH = os.path.join(PYVISI_TEST_DATA_ROOT,"data_sample_images","carpet") |
17 |
MIN_IMAGE_SIZE = 100 |
MIN_IMAGE_SIZE = 100 |
18 |
FILE_3D = "interior_3D.xml" |
FILE_3D = "interior_3D.xml" |
19 |
|
|
43 |
|
|
44 |
self.data_collector = DataCollector(source = Source.XML) |
self.data_collector = DataCollector(source = Source.XML) |
45 |
self.data_collector.setFileName(file_name = \ |
self.data_collector.setFileName(file_name = \ |
46 |
PYVISI_TEST_MESHES_PATH + FILE_3D) |
os.path.join(PYVISI_TEST_MESHES_PATH, FILE_3D)) |
47 |
|
|
48 |
self.carpet = Carpet(scene = self.scene, |
self.carpet = Carpet(scene = self.scene, |
49 |
data_collector = self.data_collector, |
data_collector = self.data_collector, |
54 |
def testScalarWarp(self): |
def testScalarWarp(self): |
55 |
self.carpet.setScaleFactor(1) |
self.carpet.setScaleFactor(1) |
56 |
self.carpet.setPlaneToXY() |
self.carpet.setPlaneToXY() |
57 |
self.render("TestCarpetScalarWarp.jpg") |
self.render(os.path.join(PYVISI_WORKDIR,"TestCarpetScalarWarp.jpg")) |
58 |
|
|
59 |
class TestCarpetVectorWarp(unittest.TestCase, TestCarpet): |
class TestCarpetVectorWarp(unittest.TestCase, TestCarpet): |
60 |
def setUp(self): |
def setUp(self): |
64 |
|
|
65 |
self.data_collector = DataCollector(source = Source.XML) |
self.data_collector = DataCollector(source = Source.XML) |
66 |
self.data_collector.setFileName(file_name = \ |
self.data_collector.setFileName(file_name = \ |
67 |
PYVISI_TEST_MESHES_PATH + FILE_3D) |
os.path.join(PYVISI_TEST_MESHES_PATH, FILE_3D)) |
68 |
|
|
69 |
self.carpet = Carpet(scene = self.scene, |
self.carpet = Carpet(scene = self.scene, |
70 |
data_collector = self.data_collector, |
data_collector = self.data_collector, |
75 |
def testVectorWarp(self): |
def testVectorWarp(self): |
76 |
self.carpet.setScaleFactor(1) |
self.carpet.setScaleFactor(1) |
77 |
self.carpet.setPlaneToXY() |
self.carpet.setPlaneToXY() |
78 |
self.render("TestCarpetVectorWarp.jpg") |
self.render(os.path.join(PYVISI_WORKDIR,"TestCarpetVectorWarp.jpg")) |
79 |
|
|
80 |
|
|
81 |
|
|