52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
|
#!/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())
|
||
|
|
|