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)))
