diff --git a/unittest/CMakeLists.txt b/unittest/CMakeLists.txt index f61b4c0dfadc252c131bbdfca7834ab2b710032c..28f45470e77abdf4e59b4674980c354161fb27fb 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 0000000000000000000000000000000000000000..a68e6fe360623190eb2fbab8cd8cb32bde9fe9ed --- /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)