From 070d223d546f96aafab9411e36dd93204a95091b Mon Sep 17 00:00:00 2001
From: Justin Carpentier <justin.carpentier@inria.fr>
Date: Fri, 28 Jun 2019 13:16:27 +0200
Subject: [PATCH] test: add test for the dimensions checking
---
unittest/CMakeLists.txt | 1 +
unittest/python/test_dimensions.py | 14 ++++++++++++++
2 files changed, 15 insertions(+)
create mode 100644 unittest/python/test_dimensions.py
diff --git a/unittest/CMakeLists.txt b/unittest/CMakeLists.txt
index 723247bf..84ce752b 100644
--- a/unittest/CMakeLists.txt
+++ b/unittest/CMakeLists.txt
@@ -36,3 +36,4 @@ ENDIF()
ADD_PYTHON_UNIT_TEST("py-matrix" "unittest/python/test_matrix.py" "unittest")
ADD_PYTHON_UNIT_TEST("py-geometry" "unittest/python/test_geometry.py" "unittest")
ADD_PYTHON_UNIT_TEST("py-switch" "unittest/python/test_switch.py" "python/eigenpy")
+ADD_PYTHON_UNIT_TEST("py-dimensions" "unittest/python/test_dimensions.py" "python/eigenpy")
diff --git a/unittest/python/test_dimensions.py b/unittest/python/test_dimensions.py
new file mode 100644
index 00000000..3ccc354b
--- /dev/null
+++ b/unittest/python/test_dimensions.py
@@ -0,0 +1,14 @@
+from __future__ import print_function
+
+import eigenpy
+import numpy as np
+
+quat = eigenpy.Quaternion()
+# By default, we convert as numpy.matrix
+coeffs_vector = quat.coeffs()
+assert len(coeffs_vector.shape) == 2
+
+# Switch to numpy.array
+eigenpy.switchToNumpyArray()
+coeffs_vector = quat.coeffs()
+assert len(coeffs_vector.shape) == 1
--
GitLab