Update to 2.0.0 tree from current Fremantle build

[opencv] / tests / swig_python / roots_tests.py

diff --git a/tests/swig_python/roots_tests.py b/tests/swig_python/roots_tests.py
new file mode 100755 (executable)
index 0000000..5151b03
--- /dev/null
+++ b/tests/swig_python/roots_tests.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+
+# 2009-01-16, Xavier Delacour <xavier.delacour@gmail.com>
+
+import unittest
+from numpy import *;
+from numpy.linalg import *;
+import sys;
+
+import cvtestutils
+from cv import *;
+from adaptors import *;
+
+## these are mostly to test bindings, since there are more/better tests in tests/cxcore
+
+def verify(x,y):
+    x = Ipl2NumPy(x);
+    assert(all(abs(x - y)<1e-4));
+
+class roots_test(unittest.TestCase):
+    
+    def test_cvSolvePoly(self):
+        
+        verify(cvSolvePoly(asmatrix([-1,1]).astype(float64)),
+                    array([[(1.000000, 0.000000)]]));
+
+        verify(cvSolvePoly(asmatrix([-1,1]).astype(float32)),
+                    array([[(1.000000, 0.000000)]]));
+
+        verify(cvSolvePoly(asmatrix([-1,0,0,0,0,1]).astype(float64)),
+               array([[(1, 0)],[(0.309017, 0.951057)],[(0.309017, -0.951057)],
+                      [(-0.809017, 0.587785)],[(-0.809017, -0.587785)]]))
+
+        verify(cvSolvePoly(asmatrix([-1,0,0,0,0,1]).astype(float32)),
+               array([[(1, 0)],[(0.309017, 0.951057)],[(0.309017, -0.951057)],
+                      [(-0.809017, 0.587785)],[(-0.809017, -0.587785)]]))
+
+    def test_cvSolveCubic(self):
+
+        verify(cvSolveCubic(asmatrix([-1,0,0,1]).astype(float32))[1],
+               array([[1],[0],[0]]));
+
+        verify(cvSolveCubic(asmatrix([-1,0,0,1]).astype(float64))[1],
+               array([[1],[0],[0]]));
+
+def suite():
+    return unittest.TestLoader().loadTestsFromTestCase(roots_test)
+
+if __name__ == '__main__':
+    unittest.TextTestRunner(verbosity=2).run(suite())
+