diff options
| author | Marcus Müller <marcus@hostalia.de> | 2014-04-07 22:00:39 +0200 |
|---|---|---|
| committer | Marcus Müller <marcus@hostalia.de> | 2014-04-07 22:00:39 +0200 |
| commit | c201f977d25c5df0834ce2486b707e3df22a4f95 (patch) | |
| tree | 78f5f86ccda71bca5106f76100e3ec9d95b57eac /gnuradio-runtime/python/pmt/qa_pmt_to_python.py | |
| parent | 29a6b80f2f3b79fc27829d962475e2bbaf864fdf (diff) | |
pmt_to_python: numpy_to_uvector and reverse works, QA added
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() |