diff --git a/unittest/python/test_switch.py b/unittest/python/test_switch.py index a68e6fe360623190eb2fbab8cd8cb32bde9fe9ed..da08b1d8160b7a74a1a0af3a6a58206c6d8cad36 100644 --- a/unittest/python/test_switch.py +++ b/unittest/python/test_switch.py @@ -3,12 +3,14 @@ from __future__ import print_function import eigenpy import numpy as np +eigenpy.switchToNumpyMatrix() 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) +assert eigenpy.getNumpyType() == np.matrix # Switch to numpy.array eigenpy.switchToNumpyArray() @@ -16,3 +18,4 @@ coeffs_array = quat.coeffs() print(type(coeffs_array)) assert isinstance(coeffs_vector,np.ndarray) +assert eigenpy.getNumpyType() == np.ndarray