Vladimír Vondruš
7 years ago
7 changed files with 645 additions and 59 deletions
@ -0,0 +1,338 @@ |
|||||||
|
# |
||||||
|
# This file is part of Magnum. |
||||||
|
# |
||||||
|
# Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019 |
||||||
|
# Vladimír Vondruš <mosra@centrum.cz> |
||||||
|
# |
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a |
||||||
|
# copy of this software and associated documentation files (the "Software"), |
||||||
|
# to deal in the Software without restriction, including without limitation |
||||||
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||||
|
# and/or sell copies of the Software, and to permit persons to whom the |
||||||
|
# Software is furnished to do so, subject to the following conditions: |
||||||
|
# |
||||||
|
# The above copyright notice and this permission notice shall be included |
||||||
|
# in all copies or substantial portions of the Software. |
||||||
|
# |
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||||
|
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||||
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||||
|
# DEALINGS IN THE SOFTWARE. |
||||||
|
# |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from magnum import * |
||||||
|
from magnum import math |
||||||
|
|
||||||
|
import numpy as np |
||||||
|
|
||||||
|
class Vector(unittest.TestCase): |
||||||
|
def test_from_numpy(self): |
||||||
|
a = Vector3(np.array([1.0, 2.0, 3.0])) |
||||||
|
self.assertEqual(a, Vector3(1.0, 2.0, 3.0)) |
||||||
|
|
||||||
|
def test_to_numpy(self): |
||||||
|
a = np.array(Vector3(1.0, 2.0, 3.0)) |
||||||
|
np.testing.assert_array_equal(a, np.array([1.0, 2.0, 3.0])) |
||||||
|
|
||||||
|
def test_from_numpy_implicit(self): |
||||||
|
# This works even w/o buffer protocol |
||||||
|
a = Vector4() |
||||||
|
a.xyz = np.array([1.0, 2.0, 3.0]) |
||||||
|
|
||||||
|
b = Matrix4.translation(np.array([1.0, 2.0, 3.0])) |
||||||
|
self.assertEqual(b._translation, Vector3(1.0, 2.0, 3.0)) |
||||||
|
|
||||||
|
def test_from_numpy_implicit_typed(self): |
||||||
|
# But this doesn't, works only if buffer protocol is defined |
||||||
|
a = Vector4() |
||||||
|
a.xyz = np.array([1.0, 2.0, 3.0], dtype='float32') |
||||||
|
|
||||||
|
a = Matrix4.translation(np.array([1.0, 2.0, 3.0], dtype='float32')) |
||||||
|
self.assertEqual(a._translation, Vector3(1.0, 2.0, 3.0)) |
||||||
|
|
||||||
|
def test_from_numpy_invalid_dimensions(self): |
||||||
|
a = np.array([[1, 2], [3, 4]]) |
||||||
|
self.assertEqual(a.ndim, 2) |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "expected 1 dimension but got 2"): |
||||||
|
b = Vector3i(a) |
||||||
|
|
||||||
|
def test_from_numpy_invalid_size(self): |
||||||
|
a = np.array([1.0, 2.0, 3.0]) |
||||||
|
self.assertEqual(a.shape[0], 3) |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "expected 2 elements but got 3"): |
||||||
|
b = Vector2(a) |
||||||
|
|
||||||
|
def test_type_from_numpy(self): |
||||||
|
a = Vector3i(np.array([1, 2, -3], dtype='int32')) |
||||||
|
self.assertEqual(a, Vector3i(1, 2, -3)) |
||||||
|
|
||||||
|
a = Vector2ui(np.array([1, 2], dtype='uint32')) |
||||||
|
self.assertEqual(a, Vector2ui(1, 2)) |
||||||
|
|
||||||
|
a = Vector4i(np.array([1, 2, -3, 0], dtype='int64')) |
||||||
|
self.assertEqual(a, Vector4i(1, 2, -3, 0)) |
||||||
|
|
||||||
|
a = Vector3ui(np.array([1, 2, 3333], dtype='uint64')) |
||||||
|
self.assertEqual(a, Vector3i(1, 2, 3333)) |
||||||
|
|
||||||
|
a = Vector2d(np.array([1.0, 2.0], dtype='float32')) |
||||||
|
self.assertEqual(a, Vector2d(1.0, 2.0)) |
||||||
|
|
||||||
|
def test_type_from_numpy_invalid_float(self): |
||||||
|
a = np.array([1, 2, 3]) |
||||||
|
self.assertEqual(a.dtype, 'int64') |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "unexpected format l for a f vector"): |
||||||
|
b = Vector3(a) |
||||||
|
|
||||||
|
def test_type_from_numpy_invalid_signed(self): |
||||||
|
a = np.array([1.0, 2.0, 3.0]) |
||||||
|
self.assertEqual(a.dtype, 'float64') |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "unexpected format d for a i vector"): |
||||||
|
b = Vector3i(a) |
||||||
|
|
||||||
|
class Matrix(unittest.TestCase): |
||||||
|
def test_from_numpy(self): |
||||||
|
a = Matrix2x3(np.array( |
||||||
|
[[1.0, 2.0], |
||||||
|
[4.0, 5.0], |
||||||
|
[7.0, 8.0]])) |
||||||
|
self.assertEqual(a, Matrix2x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0))) |
||||||
|
|
||||||
|
a = Matrix3x3d(np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
self.assertEqual(a, Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
a = Matrix4x2(np.array( |
||||||
|
[[1.0, 2.0, 3.0, 4.0], |
||||||
|
[5.0, 6.0, 7.0, 8.0]])) |
||||||
|
self.assertEqual(a, Matrix4x2( |
||||||
|
(1.0, 5.0), |
||||||
|
(2.0, 6.0), |
||||||
|
(3.0, 7.0), |
||||||
|
(4.0, 8.0))) |
||||||
|
|
||||||
|
def test_to_numpy(self): |
||||||
|
a = np.array(Matrix2x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0))) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0], |
||||||
|
[4.0, 5.0], |
||||||
|
[7.0, 8.0]])) |
||||||
|
|
||||||
|
a = np.array(Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
a = np.array(Matrix4x2( |
||||||
|
(1.0, 5.0), |
||||||
|
(2.0, 6.0), |
||||||
|
(3.0, 7.0), |
||||||
|
(4.0, 8.0))) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0, 4.0], |
||||||
|
[5.0, 6.0, 7.0, 8.0]])) |
||||||
|
|
||||||
|
def test_from_numpy_invalid_dimensions(self): |
||||||
|
a = np.array([1, 2, 3, 4]) |
||||||
|
self.assertEqual(a.ndim, 1) |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "expected 2 dimensions but got 1"): |
||||||
|
b = Matrix2x2(a) |
||||||
|
|
||||||
|
def test_from_numpy_invalid_size(self): |
||||||
|
a = np.array([[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0]]) |
||||||
|
self.assertEqual(a.shape[0], 2) |
||||||
|
self.assertEqual(a.shape[1], 3) |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "expected 2x3 elements but got 3x2"): |
||||||
|
b = Matrix2x3(a) |
||||||
|
|
||||||
|
def test_order_from_numpy(self): |
||||||
|
a = np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]]) |
||||||
|
self.assertEqual(a.strides[0], 24) |
||||||
|
self.assertEqual(a.strides[1], 8) |
||||||
|
self.assertEqual(Matrix3x3d(a), Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
a = np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]], order='C') |
||||||
|
self.assertEqual(a.strides[0], 24) |
||||||
|
self.assertEqual(a.strides[1], 8) |
||||||
|
self.assertEqual(Matrix3x3d(a), Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
a = np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]], order='F') |
||||||
|
self.assertEqual(a.strides[0], 8) |
||||||
|
self.assertEqual(a.strides[1], 24) |
||||||
|
self.assertEqual(Matrix3x3d(a), Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
def test_order_to_numpy(self): |
||||||
|
a = np.array(Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
self.assertEqual(a.strides[0], 4) |
||||||
|
self.assertEqual(a.strides[1], 12) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
a = np.array(Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0)), order='C') |
||||||
|
self.assertEqual(a.strides[0], 12) |
||||||
|
self.assertEqual(a.strides[1], 4) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
a = np.array(Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0)), order='F') |
||||||
|
self.assertEqual(a.strides[0], 4) |
||||||
|
self.assertEqual(a.strides[1], 12) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
def test_type_from_numpy(self): |
||||||
|
a = np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]], dtype='f') |
||||||
|
self.assertEqual(a.dtype, 'f') |
||||||
|
self.assertEqual(a.itemsize, 4) |
||||||
|
self.assertEqual(Matrix3x3d(a), Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
a = np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]], dtype='d') |
||||||
|
self.assertEqual(a.dtype, 'd') |
||||||
|
self.assertEqual(a.itemsize, 8) |
||||||
|
self.assertEqual(Matrix3x3(a), Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
def test_type_from_numpy_invalid(self): |
||||||
|
a = np.array([[1, 2], [3, 4]]) |
||||||
|
self.assertEqual(a.dtype, 'int64') |
||||||
|
|
||||||
|
with self.assertRaisesRegex(BufferError, "expected format f or d but got l"): |
||||||
|
b = Matrix2x2(a) |
||||||
|
|
||||||
|
def test_type_to_numpy(self): |
||||||
|
a = np.array(Matrix3x3d( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0)), dtype='f') |
||||||
|
self.assertEqual(a.dtype, 'f') |
||||||
|
self.assertEqual(a.itemsize, 4) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
a = np.array(Matrix3x3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0)), dtype='d') |
||||||
|
self.assertEqual(a.dtype, 'd') |
||||||
|
self.assertEqual(a.itemsize, 8) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
class Matrix3_(unittest.TestCase): |
||||||
|
def test_from_numpy(self): |
||||||
|
a = Matrix3(np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
self.assertEqual(a, Matrix3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
|
||||||
|
def test_to_numpy(self): |
||||||
|
a = np.array(Matrix3( |
||||||
|
(1.0, 4.0, 7.0), |
||||||
|
(2.0, 5.0, 8.0), |
||||||
|
(3.0, 6.0, 9.0))) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0], |
||||||
|
[4.0, 5.0, 6.0], |
||||||
|
[7.0, 8.0, 9.0]])) |
||||||
|
|
||||||
|
class Matrix4_(unittest.TestCase): |
||||||
|
def test_from_numpy(self): |
||||||
|
a = Matrix4(np.array( |
||||||
|
[[1.0, 2.0, 3.0, 4.0], |
||||||
|
[5.0, 6.0, 7.0, 8.0], |
||||||
|
[9.0, 10.0, 11.0, 12.0], |
||||||
|
[13.0, 14.0, 15.0, 16.0]])) |
||||||
|
self.assertEqual(a, Matrix4( |
||||||
|
(1.0, 5.0, 9.0, 13.0), |
||||||
|
(2.0, 6.0, 10.0, 14.0), |
||||||
|
(3.0, 7.0, 11.0, 15.0), |
||||||
|
(4.0, 8.0, 12.0, 16.0))) |
||||||
|
|
||||||
|
def test_to_numpy(self): |
||||||
|
a = np.array(Matrix4( |
||||||
|
(1.0, 5.0, 9.0, 13.0), |
||||||
|
(2.0, 6.0, 10.0, 14.0), |
||||||
|
(3.0, 7.0, 11.0, 15.0), |
||||||
|
(4.0, 8.0, 12.0, 16.0))) |
||||||
|
np.testing.assert_array_equal(a, np.array( |
||||||
|
[[1.0, 2.0, 3.0, 4.0], |
||||||
|
[5.0, 6.0, 7.0, 8.0], |
||||||
|
[9.0, 10.0, 11.0, 12.0], |
||||||
|
[13.0, 14.0, 15.0, 16.0]])) |
||||||
@ -0,0 +1,59 @@ |
|||||||
|
# |
||||||
|
# This file is part of Magnum. |
||||||
|
# |
||||||
|
# Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019 |
||||||
|
# Vladimír Vondruš <mosra@centrum.cz> |
||||||
|
# |
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a |
||||||
|
# copy of this software and associated documentation files (the "Software"), |
||||||
|
# to deal in the Software without restriction, including without limitation |
||||||
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||||
|
# and/or sell copies of the Software, and to permit persons to whom the |
||||||
|
# Software is furnished to do so, subject to the following conditions: |
||||||
|
# |
||||||
|
# The above copyright notice and this permission notice shall be included |
||||||
|
# in all copies or substantial portions of the Software. |
||||||
|
# |
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||||
|
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||||
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||||
|
# DEALINGS IN THE SOFTWARE. |
||||||
|
# |
||||||
|
|
||||||
|
import numpy as np |
||||||
|
import unittest |
||||||
|
|
||||||
|
from magnum import * |
||||||
|
from magnum import scenegraph |
||||||
|
from magnum.scenegraph.matrix import Object3D, Scene3D |
||||||
|
|
||||||
|
class Object(unittest.TestCase): |
||||||
|
def test_transformation(self): |
||||||
|
scene = Scene3D() |
||||||
|
|
||||||
|
a = Object3D(scene) |
||||||
|
|
||||||
|
# like a.rotate_local(Deg(35.0), Vector3.x_axis()), but way uglier, |
||||||
|
# another could be scipy.spatial.transform.Rotation but that's meh as |
||||||
|
# well |
||||||
|
a.transform_local(np.array( |
||||||
|
[[1.0, 0.0, 0.0, 0.0], |
||||||
|
[0.0, 0.819152, -0.573576, 0.0], |
||||||
|
[0.0, 0.573576, 0.819152, 0.0], |
||||||
|
[0.0, 0.0, 0.0, 1.0]])) |
||||||
|
self.assertEqual(a.transformation, Matrix4.rotation_x(Deg(35.0))) |
||||||
|
self.assertEqual(a.absolute_transformation(), Matrix4.rotation_x(Deg(35.0))) |
||||||
|
|
||||||
|
b = Object3D(a) |
||||||
|
b.translate(np.array([3.0, 4.0, 5.0], dtype='float32')) |
||||||
|
self.assertEqual(b.transformation, Matrix4.translation((3.0, 4.0, 5.0))) |
||||||
|
self.assertEqual(b.absolute_transformation(), |
||||||
|
Matrix4.rotation_x(Deg(35.0))@ |
||||||
|
Matrix4.translation((3.0, 4.0, 5.0))) |
||||||
|
|
||||||
|
c = Object3D(scene) |
||||||
|
self.assertEqual(c.transformation, Matrix4.identity_init()) |
||||||
|
self.assertEqual(c.absolute_transformation(), Matrix4.identity_init()) |
||||||
Loading…
Reference in new issue