fix tests

stack-of-tasks · May 12, 2023 · 901a479 · 901a479
1 parent 9ffea93
commit 901a479
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 12 deletions.
diff --git a/python/spline_python.cpp b/python/spline_python.cpp
@@ -244,7 +244,7 @@ void set_end_acc(curve_constraints_t& c, const point_t& val) {
 
 BOOST_PYTHON_MODULE(libparametric_curves_pywrap) {
   /** BEGIN eigenpy init**/
-  // eigenpy::enableEigenPy();
+  bp::import("eigenpy");
 
   eigenpy::enableEigenPySpecific<point_t>();
   eigenpy::enableEigenPySpecific<ret_point_t>();

diff --git a/tests/python/CMakeLists.txt b/tests/python/CMakeLists.txt
@@ -1,4 +1,3 @@
 # ADD_PYTHON_UNIT_TEST("py-spline" "tests/python/test.py"
 # "python/parametric_curves")
-add_python_unit_test("py-quick" "tests/python/quick.py"
-                     "python/parametric_curves")
+add_python_unit_test("py-quick" "tests/python/quick.py" "python")
diff --git a/tests/python/quick.py b/tests/python/quick.py
@@ -1,23 +1,17 @@
 import unittest
 
 from numpy import array
-
-from libparametric_curves_pywrap import (
-    curve_constraints,
-    forcecurve,
-    polynomial,
-    spline,
-)
+from parametric_curves import curve_constraints, forcecurve, polynomial, spline
 
 
 class ParametricCurvesQuickTests(unittest.TestCase):
     def test_quick(self):
         cc = curve_constraints()
-        cc.init_vel = array(range(3))
+        cc.init_vel = array([0, 1, 2])
 
         forcecurve()
         spline()
-        pn = polynomial(array("1 2 3;4 5 6;7 8 9"))
+        pn = polynomial(array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
 
         self.assertEqual(pn.min(), 0.0)
         self.assertEqual(pn.max(), 1.0)