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_vector4(self): v1 = Vector4() v2 = Vector4([1., 2., 3., 4.]) # add self.assertTrue(np.array_equal(v1 + v2, [1., 2., 3., 4.])) # subtract self.assertTrue(np.array_equal(v1 - v2, [-1., -2., -3., -4])) # multiply self.assertTrue(np.array_equal(v1 * v2, [0., 0., 0., 0.])) # divide self.assertTrue(np.array_equal(v1 / v2, [0., 0., 0., 0.])) # or self.assertTrue( np.array_equal(v1 | v2, vector4.dot([0., 0., 0., 0.], [1., 2., 3., 4.]))) # xor #self.assertTrue(np.array_equal(v1 ^ v2, vector4.cross([0.,0.,0.,0.], [1.,2.,3.,4.]))) # == self.assertTrue(Vector4() == Vector4()) self.assertFalse(Vector4() == Vector4([1., 1., 1., 1.])) # != self.assertTrue(Vector4() != Vector4([1., 1., 1., 1.])) self.assertFalse(Vector4() != Vector4())
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_operators_number(self): v1 = Vector4([1., 2., 3., 4.]) 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., 5.])) self.assertTrue(np.array_equal(v1 + 1, [2., 3., 4., 5.])) self.assertTrue(np.array_equal(v1 + np.float(1.), [2., 3., 4., 5.])) self.assertTrue(np.array_equal(v1 + fv[0]['f'], [3., 4., 5., 6.])) self.assertTrue(np.array_equal(v1 + fv[0]['i'], [3., 4., 5., 6.])) # subtract self.assertTrue(np.array_equal(v1 - 1., [0., 1., 2., 3.])) self.assertTrue(np.array_equal(v1 - 1, [0., 1., 2., 3.])) self.assertTrue(np.array_equal(v1 - np.float(1.), [0., 1., 2., 3.])) self.assertTrue(np.array_equal(v1 - fv[0]['f'], [-1., 0., 1., 2.])) self.assertTrue(np.array_equal(v1 - fv[0]['i'], [-1., 0., 1., 2.])) # multiply self.assertTrue(np.array_equal(v1 * 2., [2., 4., 6., 8.])) self.assertTrue(np.array_equal(v1 * 2, [2., 4., 6., 8.])) self.assertTrue(np.array_equal(v1 * np.float(2.), [2., 4., 6., 8.])) self.assertTrue(np.array_equal(v1 * fv[0]['f'], [2., 4., 6., 8.])) self.assertTrue(np.array_equal(v1 * fv[0]['i'], [2., 4., 6., 8.])) # divide self.assertTrue(np.array_equal(v1 / 2., [.5, 1., 1.5, 2.])) self.assertTrue(np.array_equal(v1 / 2, [.5, 1., 1.5, 2.])) self.assertTrue(np.array_equal(v1 / np.float(2.), [.5, 1., 1.5, 2.])) self.assertTrue(np.array_equal(v1 / fv[0]['f'], [.5, 1., 1.5, 2.])) self.assertTrue(np.array_equal(v1 / fv[0]['i'], [.5, 1., 1.5, 2.])) # or self.assertRaises(ValueError, lambda: v1 | .5) self.assertRaises(ValueError, lambda: v1 | 5) self.assertRaises(ValueError, lambda: v1 | np.float(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.float(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.float(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.float(2.)) self.assertRaises(ValueError, lambda: v1 != fv[0]['f']) self.assertRaises(ValueError, lambda: v1 != fv[0]['i'])
def test_normalize(self): v = Vector4([1., 1., 1., 1.]) np.testing.assert_almost_equal(v.normalized, [0.5, 0.5, 0.5, 0.5], decimal=5) v.normalize() np.testing.assert_almost_equal(v, [0.5, 0.5, 0.5, 0.5], decimal=5)
def test_operators_quaternion(self): v = Vector4() 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_matrix44(self): v = Vector4() 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_create(self): v = Vector4() self.assertTrue(np.array_equal(v, [0.,0.,0.,0.])) self.assertEqual(v.shape, self._shape) v = Vector4([1.,2.,3.,4.]) self.assertTrue(np.array_equal(v, [1.,2.,3.,4.])) self.assertEqual(v.shape, self._shape) v = Vector4.from_vector3([1.,2.,3.], w=0.0) self.assertTrue(np.array_equal(v, [1.,2.,3.,0.])) self.assertEqual(v.shape, self._shape) v = Vector4(Vector4()) self.assertTrue(np.array_equal(v, [0.,0.,0.,0.])) self.assertEqual(v.shape, self._shape) m = Matrix44.from_translation([1.,2.,3.]) v = Vector4.from_matrix44_translation(m) self.assertTrue(np.array_equal(v, [1.,2.,3.,1.])) m = Matrix44.from_translation([1.,2.,3.]) v = Vector4(m) self.assertTrue(np.array_equal(v, [1.,2.,3.,1.]))
def test_operators_vector4(self): m = Matrix33.identity() v = Vector4([1,1,1,1]) # add self.assertRaises(ValueError, lambda: m + v) # subtract self.assertRaises(ValueError, lambda: m - v) # multiply self.assertTrue(ValueError, lambda: m * v) # divide self.assertRaises(ValueError, lambda: m / v)
def test_operators_vector4(self): m = Matrix44.identity() v = Vector4([1,1,1,1]) # add self.assertRaises(ValueError, lambda: m + v) # subtract self.assertRaises(ValueError, lambda: m - v) # multiply self.assertTrue(np.array_equal(m * v, matrix44.apply_to_vector(matrix44.create_identity(), [1,1,1,1]))) # divide self.assertRaises(ValueError, lambda: m / v)
def test_operators_vector4(self): q = Quaternion.from_x_rotation(0.5) v = Vector4([1.,0.,0.,1.]) # add self.assertRaises(ValueError, lambda: q + v) # subtract self.assertRaises(ValueError, lambda: q - v) # multiply self.assertTrue(np.array_equal(q * v, quaternion.apply_to_vector(quaternion.create_from_x_rotation(0.5), [1.,0.,0.,1.]))) # divide self.assertRaises(ValueError, lambda: q / v)
def test_accessors(self): v = Vector4(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)
def test_create(self): v = Vector4() self.assertTrue(np.array_equal(v, [0., 0., 0., 0.])) self.assertEqual(v.shape, self._shape) v = Vector4([1., 2., 3., 4.]) self.assertTrue(np.array_equal(v, [1., 2., 3., 4.])) self.assertEqual(v.shape, self._shape) v = Vector4.from_vector3([1., 2., 3.], w=0.0) self.assertTrue(np.array_equal(v, [1., 2., 3., 0.])) self.assertEqual(v.shape, self._shape) v = Vector4(Vector4()) self.assertTrue(np.array_equal(v, [0., 0., 0., 0.])) self.assertEqual(v.shape, self._shape) m = Matrix44.from_translation([1., 2., 3.]) v = Vector4.from_matrix44_translation(m) self.assertTrue(np.array_equal(v, [1., 2., 3., 1.])) m = Matrix44.from_translation([1., 2., 3.]) v = Vector4(m) self.assertTrue(np.array_equal(v, [1., 2., 3., 1.]))
def test_inverse(self): v = Vector4([1., 2., 3., 4.]) self.assertTrue(np.array_equal(v.inverse, [-1., -2., -3., -4.]))
def test_bitwise(self): v1 = Vector4([1., 0., 0., 1.]) v2 = Vector4([0., 1., 0., 1.]) # or (dot) self.assertTrue(np.array_equal(v1 | v2, vector4.dot(v1, v2)))