From 28d585b29c1d54f615ad899ddfc0177f4a2a1135 Mon Sep 17 00:00:00 2001
From: Justin Carpentier <justin.carpentier@inria.fr>
Date: Thu, 18 Oct 2018 18:46:35 +0200
Subject: [PATCH] test: add test on switch between np.array and np.matrix
---
unittest/CMakeLists.txt | 3 ++-
unittest/python/test_switch.py | 18 ++++++++++++++++++
2 files changed, 20 insertions(+), 1 deletion(-)
create mode 100644 unittest/python/test_switch.py
diff --git a/unittest/CMakeLists.txt b/unittest/CMakeLists.txt
index f61b4c0d..28f45470 100644
--- a/unittest/CMakeLists.txt
+++ b/unittest/CMakeLists.txt
@@ -1,5 +1,5 @@
#
-# Copyright (c) 2016-2018 CNRS
+# Copyright (c) 2016-2018 CNRS INRIA
#
# This file is part of eigenpy
# eigenpy is free software: you can redistribute it
@@ -46,3 +46,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" "unittest")
diff --git a/unittest/python/test_switch.py b/unittest/python/test_switch.py
new file mode 100644
index 00000000..a68e6fe3
--- /dev/null
+++ b/unittest/python/test_switch.py
@@ -0,0 +1,18 @@
+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()
+print(type(coeffs_vector))
+
+assert isinstance(coeffs_vector,np.matrixlib.defmatrix.matrix)
+
+# Switch to numpy.array
+eigenpy.switchToNumpyArray()
+coeffs_array = quat.coeffs()
+print(type(coeffs_array))
+
+assert isinstance(coeffs_vector,np.ndarray)
--
GitLab