diff options
| author | Tom Rondeau <tom@trondeau.com> | 2014-04-10 13:45:43 -0400 |
|---|---|---|
| committer | Tom Rondeau <tom@trondeau.com> | 2014-04-10 13:45:43 -0400 |
| commit | 071c36265f97e0a13b00564b1f46e75be1e1522f (patch) | |
| tree | 84684e9b2be005dd1fc43fc85907ff9ffd5b2086 /gnuradio-runtime/python/pmt/qa_pmt_to_python.py | |
| parent | 9880ba4cd98925c9e5803847ff22af8038514fba (diff) | |
| parent | 10c48ced05c22266ae089ac560b2beb45f67851f (diff) | |
Merge remote-tracking branch 'mmueller/pmt_to_python-py2.6-compat'
Diffstat (limited to 'gnuradio-runtime/python/pmt/qa_pmt_to_python.py')
| -rwxr-xr-x | gnuradio-runtime/python/pmt/qa_pmt_to_python.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/gnuradio-runtime/python/pmt/qa_pmt_to_python.py b/gnuradio-runtime/python/pmt/qa_pmt_to_python.py index ae86fc6d53..39cfc05dd6 100755 --- a/gnuradio-runtime/python/pmt/qa_pmt_to_python.py +++ b/gnuradio-runtime/python/pmt/qa_pmt_to_python.py @@ -22,13 +22,27 @@ import unittest import pmt +import pmt_to_python as pmt2py class test_pmt_to_python(unittest.TestCase): - def test01 (self): + def test_pmt_from_double(self): b = pmt.from_double(123765) self.assertEqual(pmt.to_python(b), 123765) t = pmt.to_pmt(range(5)) + + def test_numpy_to_uvector_and_reverse(self): + import numpy as np + N = 100 + narr = np.ndarray(N, dtype=np.complex128) + narr.real[:] = np.random.uniform(size=N) + narr.imag[:] = np.random.uniform(size=N) + uvector = pmt2py.numpy_to_uvector(narr) + nparr = pmt2py.uvector_to_numpy(uvector) + self.assertTrue(nparr.dtype==narr.dtype) + self.assertTrue(np.alltrue(nparr == narr)) + + if __name__ == '__main__': unittest.main() |