from __future__ import absolute_import try: import unittest2 as unittest except: import unittest import numpy as np from pyrr.objects.matrix33 import Matrix33 from pyrr.objects.matrix44 import Matrix44 from pyrr.objects.quaternion import Quaternion from pyrr.objects.vector3 import Vector3 from pyrr.objects.vector4 import Vector4 from pyrr import vector3 class test_object_vector3(unittest.TestCase): _shape = (3,) _size = np.multiply.reduce(_shape) def test_imports(self): import pyrr pyrr.Vector3() pyrr.vector3.Vector3() pyrr.objects.vector3.Vector3() def test_imports_1(self): from pyrr import Vector3 Vector3 def test_imports_2(self): from pyrr.objects import Vector3 Vector3 def test_imports_3(self): from pyrr.objects.vector3 import Vector3 Vector3 def test_create(self): v = Vector3() self.assertTrue(np.array_equal(v, [0.,0.,0.])) self.assertEqual(v.shape, self._shape) v = Vector3([1.,2.,3.]) self.assertTrue(np.array_equal(v, [1.,2.,3.])) self.assertEqual(v.shape, self._shape) v = Vector3(Vector3()) self.assertTrue(np.array_equal(v, [0.,0.,0.])) self.assertEqual(v.shape, self._shape) v4 = [1., 2., 3., 4.] result = vector3.create_from_vector4(v4) v, w = result np.testing.assert_almost_equal(v, [1.,2.,3.], decimal=5) np.testing.assert_almost_equal(w, 4., decimal=5) v4 = Vector4([1., 2., 3., 4.]) result = vector3.create_from_vector4(v4) v, w = result np.testing.assert_almost_equal(v, [1.,2.,3.], decimal=5) np.testing.assert_almost_equal(w, 4., decimal=5) m = Matrix44.from_translation([1.,2.,3.]) v = Vector3.from_matrix44_translation(m) self.assertTrue(np.array_equal(v, [1.,2.,3.])) m = Matrix44.from_translation([1.,2.,3.]) v = Vector3(m) self.assertTrue(np.array_equal(v, [1.,2.,3.])) def test_inverse(self): v = Vector3([1.,2.,3.]) self.assertTrue(np.array_equal(v.inverse, [-1.,-2.,-3.])) def test_normalize(self): v = Vector3([1.,1.,1.]) np.testing.assert_almost_equal(np.array(v.normalized), [0.57735, 0.57735, 0.57735], decimal=5) v.normalize() np.testing.assert_almost_equal(np.array(v), [0.57735, 0.57735, 0.57735], decimal=5) def test_operators_matrix33(self): v = Vector3() m = Matrix33.from_x_rotation(0.5) # add self.assertRaises(ValueError, lambda: v + m) # subtract self.assertRaises(ValueError, lambda: v - m) # multiply self.assertRaises(ValueError, lambda: v - m) # divide self.assertRaises(ValueError, lambda: v / m) def test_operators_matrix44(self): v = Vector3() m = Matrix44.from_x_rotation(0.5) # add self.assertRaises(ValueError, lambda: v + m) # subtract self.assertRaises(ValueError, lambda: v - m) # multiply self.assertRaises(ValueError, lambda: v * m) # divide self.assertRaises(ValueError, lambda: v / m) def test_operators_quaternion(self): v = Vector3() q = Quaternion.from_x_rotation(0.5) # add self.assertRaises(ValueError, lambda: v + q) # subtract self.assertRaises(ValueError, lambda: v - q) # multiply self.assertRaises(ValueError, lambda: v * q) # divide self.assertRaises(ValueError, lambda: v / q) def test_operators_vector3(self): v1 = Vector3() v2 = Vector3([1.,2.,3.]) # add self.assertTrue(np.array_equal(v1 + v2, [1.,2.,3.])) # subtract self.assertTrue(np.array_equal(v1 - v2, [-1.,-2.,-3.])) # multiply self.assertTrue(np.array_equal(v1 * v2, [0.,0.,0.])) # divide self.assertTrue(np.array_equal(v1 / v2, [0.,0.,0.])) # or self.assertTrue(np.array_equal(v1 | v2, vector3.dot(v1, v2))) # xor self.assertTrue(np.array_equal(v1 ^ v2, vector3.cross(v1, v2))) # == self.assertTrue(Vector3() == Vector3()) self.assertFalse(Vector3() == Vector3([1.,1.,1.])) # != self.assertTrue(Vector3() != Vector3([1.,1.,1.])) self.assertFalse(Vector3() != Vector3()) def test_operators_vector4(self): v1 = Vector3() v2 = Vector4([1.,2.,3.,4.]) # add self.assertRaises(ValueError, lambda: v1 + v2) # subtract self.assertRaises(ValueError, lambda: v1 - v2) # multiply self.assertRaises(ValueError, lambda: v1 * v2) # divide self.assertRaises(ValueError, lambda: v1 / v2) # or self.assertRaises(ValueError, lambda: v1 | v2) # xor self.assertRaises(ValueError, lambda: v1 ^ v2) # == self.assertRaises(ValueError, lambda: Vector3() == Vector4()) # != self.assertRaises(ValueError, lambda: Vector3() != Vector4([1.,1.,1.,1.])) def test_operators_number(self): v1 = Vector3([1.,2.,3.]) fv = np.empty((1,), dtype=[('i', np.int16, 1),('f', np.float32, 1)]) fv[0] = (2, 2.0) # add self.assertTrue(np.array_equal(v1 + 1., [2., 3., 4.])) self.assertTrue(np.array_equal(v1 + 1, [2., 3., 4.])) self.assertTrue(np.array_equal(v1 + np.float64(1.), [2., 3., 4.])) self.assertTrue(np.array_equal(v1 + fv[0]['f'], [3., 4., 5.])) self.assertTrue(np.array_equal(v1 + fv[0]['i'], [3., 4., 5.])) # subtract self.assertTrue(np.array_equal(v1 - 1., [0., 1., 2.])) self.assertTrue(np.array_equal(v1 - 1, [0., 1., 2.])) self.assertTrue(np.array_equal(v1 - np.float64(1.), [0., 1., 2.])) self.assertTrue(np.array_equal(v1 - fv[0]['f'], [-1., 0., 1.])) self.assertTrue(np.array_equal(v1 - fv[0]['i'], [-1., 0., 1.])) # multiply self.assertTrue(np.array_equal(v1 * 2., [2., 4., 6.])) self.assertTrue(np.array_equal(v1 * 2, [2., 4., 6.])) self.assertTrue(np.array_equal(v1 * np.float64(2.), [2., 4., 6.])) self.assertTrue(np.array_equal(v1 * fv[0]['f'], [2., 4., 6.])) self.assertTrue(np.array_equal(v1 * fv[0]['i'], [2., 4., 6.])) # divide self.assertTrue(np.array_equal(v1 / 2., [.5, 1., 1.5])) self.assertTrue(np.array_equal(v1 / 2, [.5, 1., 1.5])) self.assertTrue(np.array_equal(v1 / np.float64(2.), [.5, 1., 1.5])) self.assertTrue(np.array_equal(v1 / fv[0]['f'], [.5, 1., 1.5])) self.assertTrue(np.array_equal(v1 / fv[0]['i'], [.5, 1., 1.5])) # or self.assertRaises(ValueError, lambda: v1 | .5) self.assertRaises(ValueError, lambda: v1 | 5) self.assertRaises(ValueError, lambda: v1 | np.float64(2.)) self.assertRaises(ValueError, lambda: v1 | fv[0]['f']) self.assertRaises(ValueError, lambda: v1 | fv[0]['i']) # xor self.assertRaises(ValueError, lambda: v1 ^ .5) self.assertRaises(ValueError, lambda: v1 ^ 5) self.assertRaises(ValueError, lambda: v1 ^ np.float64(2.)) self.assertRaises(ValueError, lambda: v1 ^ fv[0]['f']) self.assertRaises(ValueError, lambda: v1 ^ fv[0]['i']) # == self.assertRaises(ValueError, lambda: v1 == .5) self.assertRaises(ValueError, lambda: v1 == 5) self.assertRaises(ValueError, lambda: v1 == np.float64(2.)) self.assertRaises(ValueError, lambda: v1 == fv[0]['f']) self.assertRaises(ValueError, lambda: v1 == fv[0]['i']) # != self.assertRaises(ValueError, lambda: v1 != .5) self.assertRaises(ValueError, lambda: v1 != 5) self.assertRaises(ValueError, lambda: v1 != np.float64(2.)) self.assertRaises(ValueError, lambda: v1 != fv[0]['f']) self.assertRaises(ValueError, lambda: v1 != fv[0]['i']) def test_bitwise(self): v1 = Vector3([1.,0.,0.]) v2 = Vector3([0.,1.,0.]) # xor (cross) self.assertTrue(np.array_equal(v1 ^ v2, vector3.cross(v1, v2))) # or (dot) self.assertTrue(np.array_equal(v1 | v2, vector3.dot(v1, v2))) def test_accessors(self): v = Vector3(np.arange(self._size)) self.assertTrue(np.array_equal(v.xy,[0,1])) self.assertTrue(np.array_equal(v.xyz,[0,1,2])) self.assertTrue(np.array_equal(v.xz,[0,2])) self.assertTrue(np.array_equal(v.xyz,[0,1,2])) self.assertEqual(v.x, 0) self.assertEqual(v.y, 1) self.assertEqual(v.z, 2) v.x = 1 self.assertEqual(v.x, 1) self.assertEqual(v[0], 1) v.x += 1 self.assertEqual(v.x, 2) self.assertEqual(v[0], 2) if __name__ == '__main__': unittest.main()