1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import collada
from collada.util import unittest
from collada.xmlutil import etree
fromstring = etree.fromstring
tostring = etree.tostring
class TestLight(unittest.TestCase):
def setUp(self):
self.dummy = collada.Collada(validate_output=True)
def test_directional_light_saving(self):
dirlight = collada.light.DirectionalLight("mydirlight", (1,1,1))
self.assertEqual(dirlight.id, "mydirlight")
self.assertTupleEqual(dirlight.color, (1,1,1))
self.assertTupleEqual(tuple(dirlight.direction), (0,0,-1))
self.assertIsNotNone(str(dirlight))
dirlight.color = (0.1, 0.2, 0.3)
dirlight.id = "yourdirlight"
dirlight.save()
loaded_dirlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(dirlight.xmlnode)))
self.assertTrue(isinstance(loaded_dirlight, collada.light.DirectionalLight))
self.assertTupleEqual(loaded_dirlight.color, (0.1, 0.2, 0.3))
self.assertEqual(loaded_dirlight.id, "yourdirlight")
def test_ambient_light_saving(self):
ambientlight = collada.light.AmbientLight("myambientlight", (1,1,1))
self.assertEqual(ambientlight.id, "myambientlight")
self.assertTupleEqual(ambientlight.color, (1,1,1))
self.assertIsNotNone(str(ambientlight))
ambientlight.color = (0.1, 0.2, 0.3)
ambientlight.id = "yourambientlight"
ambientlight.save()
loaded_ambientlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(ambientlight.xmlnode)))
self.assertTrue(isinstance(loaded_ambientlight, collada.light.AmbientLight))
self.assertTupleEqual(ambientlight.color, (0.1, 0.2, 0.3))
self.assertEqual(ambientlight.id, "yourambientlight")
def test_point_light_saving(self):
pointlight = collada.light.PointLight("mypointlight", (1,1,1))
self.assertEqual(pointlight.id, "mypointlight")
self.assertTupleEqual(pointlight.color, (1,1,1))
self.assertEqual(pointlight.quad_att, None)
self.assertEqual(pointlight.constant_att, None)
self.assertEqual(pointlight.linear_att, None)
self.assertEqual(pointlight.zfar, None)
self.assertIsNotNone(str(pointlight))
pointlight.color = (0.1, 0.2, 0.3)
pointlight.constant_att = 0.7
pointlight.linear_att = 0.8
pointlight.quad_att = 0.9
pointlight.id = "yourpointlight"
pointlight.save()
loaded_pointlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(pointlight.xmlnode)))
self.assertTrue(isinstance(loaded_pointlight, collada.light.PointLight))
self.assertTupleEqual(loaded_pointlight.color, (0.1, 0.2, 0.3))
self.assertEqual(loaded_pointlight.constant_att, 0.7)
self.assertEqual(loaded_pointlight.linear_att, 0.8)
self.assertEqual(loaded_pointlight.quad_att, 0.9)
self.assertEqual(loaded_pointlight.zfar, None)
self.assertEqual(loaded_pointlight.id, "yourpointlight")
loaded_pointlight.zfar = 0.2
loaded_pointlight.save()
loaded_pointlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(loaded_pointlight.xmlnode)))
self.assertEqual(loaded_pointlight.zfar, 0.2)
def test_spot_light_saving(self):
spotlight = collada.light.SpotLight("myspotlight", (1,1,1))
self.assertEqual(spotlight.id, "myspotlight")
self.assertTupleEqual(spotlight.color, (1,1,1))
self.assertEqual(spotlight.constant_att, None)
self.assertEqual(spotlight.linear_att, None)
self.assertEqual(spotlight.quad_att, None)
self.assertEqual(spotlight.falloff_ang, None)
self.assertEqual(spotlight.falloff_exp, None)
self.assertIsNotNone(str(spotlight))
spotlight.color = (0.1, 0.2, 0.3)
spotlight.constant_att = 0.7
spotlight.linear_att = 0.8
spotlight.quad_att = 0.9
spotlight.id = "yourspotlight"
spotlight.save()
loaded_spotlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(spotlight.xmlnode)))
self.assertTrue(isinstance(loaded_spotlight, collada.light.SpotLight))
self.assertTupleEqual(loaded_spotlight.color, (0.1, 0.2, 0.3))
self.assertEqual(loaded_spotlight.constant_att, 0.7)
self.assertEqual(loaded_spotlight.linear_att, 0.8)
self.assertEqual(loaded_spotlight.quad_att, 0.9)
self.assertEqual(loaded_spotlight.falloff_ang, None)
self.assertEqual(loaded_spotlight.falloff_exp, None)
self.assertEqual(loaded_spotlight.id, "yourspotlight")
loaded_spotlight.falloff_ang = 180
loaded_spotlight.falloff_exp = 2
loaded_spotlight.save()
loaded_spotlight = collada.light.Light.load(self.dummy, {}, fromstring(tostring(loaded_spotlight.xmlnode)))
self.assertEqual(loaded_spotlight.falloff_ang, 180)
self.assertEqual(loaded_spotlight.falloff_exp, 2)
if __name__ == '__main__':
unittest.main()
|
