def test_003(self): offsets = (6, 3, 8) key = pmt.string_to_symbol('key') srcid = pmt.string_to_symbol('qa_tag_utils') tags = [] for k in offsets: t = gr.tag_t() t.offset = k t.key = key t.value = pmt.from_long(k) t.srcid = srcid tags.append(t) for k, t in zip(sorted(offsets), sorted(tags, key=gr.tag_t_offset_compare_key())): self.assertEqual(t.offset, k) self.assertTrue(pmt.equal(t.key, key)) self.assertTrue(pmt.equal(t.value, pmt.from_long(k))) self.assertTrue(pmt.equal(t.srcid, srcid)) tmin = min(tags, key=gr.tag_t_offset_compare_key()) self.assertEqual(tmin.offset, min(offsets)) self.assertTrue(pmt.equal(tmin.key, key)) self.assertTrue(pmt.equal(tmin.value, pmt.from_long(min(offsets)))) self.assertTrue(pmt.equal(tmin.srcid, srcid)) tmax = max(tags, key=gr.tag_t_offset_compare_key()) self.assertEqual(tmax.offset, max(offsets)) self.assertTrue(pmt.equal(tmax.key, key)) self.assertTrue(pmt.equal(tmax.value, pmt.from_long(max(offsets)))) self.assertTrue(pmt.equal(tmax.srcid, srcid))
def test_002(self): offset = 10 key = pmt.string_to_symbol('key') value = pmt.from_long(23) srcid = pmt.from_bool(False) format_dict = { 'offset': offset, 'key': key, 'value': value, 'srcid': srcid } format_list = [offset, key, value, srcid] format_tuple = (offset, key, value, srcid) t_dict = gr.python_to_tag(format_dict) t_list = gr.python_to_tag(format_list) t_tuple = gr.python_to_tag(format_tuple) self.assertTrue(pmt.equal(t_dict.key, key)) self.assertTrue(pmt.equal(t_dict.value, value)) self.assertEqual(t_dict.offset, offset) self.assertTrue(pmt.equal(t_list.key, key)) self.assertTrue(pmt.equal(t_list.value, value)) self.assertEqual(t_list.offset, offset) self.assertTrue(pmt.equal(t_tuple.key, key)) self.assertTrue(pmt.equal(t_tuple.value, value)) self.assertEqual(t_tuple.offset, offset)
def test_002(self): offset = 10 key = pmt.string_to_symbol("key") value = pmt.from_long(23) srcid = pmt.from_bool(False) format_dict = {"offset": offset, "key": key, "value": value, "srcid": srcid} format_list = [offset, key, value, srcid] format_tuple = (offset, key, value, srcid) t_dict = gr.python_to_tag(format_dict) t_list = gr.python_to_tag(format_list) t_tuple = gr.python_to_tag(format_tuple) self.assertTrue(pmt.equal(t_dict.key, key)) self.assertTrue(pmt.equal(t_dict.value, value)) self.assertEqual(t_dict.offset, offset) self.assertTrue(pmt.equal(t_list.key, key)) self.assertTrue(pmt.equal(t_list.value, value)) self.assertEqual(t_list.offset, offset) self.assertTrue(pmt.equal(t_tuple.key, key)) self.assertTrue(pmt.equal(t_tuple.value, value)) self.assertEqual(t_tuple.offset, offset)
def test_002(self): offset = 10 key = pmt.string_to_symbol('key') value = pmt.from_long(23) srcid = pmt.from_bool(False) format_dict = {'offset': offset, 'key': key, 'value': value, 'srcid': srcid} format_list = [offset, key, value, srcid] format_tuple = (offset, key, value, srcid) t_dict = gr.python_to_tag(format_dict) t_list = gr.python_to_tag(format_list) t_tuple = gr.python_to_tag(format_tuple) self.assertTrue(pmt.equal(t_dict.key, key)) self.assertTrue(pmt.equal(t_dict.value, value)) self.assertEqual(t_dict.offset, offset) self.assertTrue(pmt.equal(t_list.key, key)) self.assertTrue(pmt.equal(t_list.value, value)) self.assertEqual(t_list.offset, offset) self.assertTrue(pmt.equal(t_tuple.key, key)) self.assertTrue(pmt.equal(t_tuple.value, value)) self.assertEqual(t_tuple.offset, offset)
def test_001(self): t = gr.tag_t() t.offset = 10 t.key = pmt.string_to_symbol('key') t.value = pmt.from_long(23) t.srcid = pmt.from_bool(False) pt = gr.tag_to_python(t) self.assertEqual(pt.key, 'key') self.assertEqual(pt.value, 23) self.assertEqual(pt.offset, 10)
def test_001(self): t = gr.tag_t() t.offset = 10 t.key = pmt.string_to_symbol("key") t.value = pmt.from_long(23) t.srcid = pmt.from_bool(False) pt = gr.tag_to_python(t) self.assertEqual(pt.key, "key") self.assertEqual(pt.value, 23) self.assertEqual(pt.offset, 10)
return d def numpy_to_blob(p): p = p.view(numpy.uint8) b = pmt.make_blob(len(p)) pmt.blob_data(b)[:] = p return b THE_TABLE = ( #python type, check pmt type, to python, from python (None, pmt.is_null, lambda x: None, lambda x: pmt.PMT_NIL), (bool, pmt.is_bool, pmt.to_bool, pmt.from_bool), (str, pmt.is_symbol, pmt.symbol_to_string, pmt.string_to_symbol), (unicode, lambda x: False, None, lambda x: pmt.string_to_symbol(x.encode('utf-8'))), (int, pmt.is_integer, pmt.to_long, pmt.from_long), (long, pmt.is_uint64, lambda x: long(pmt.to_uint64(x)), pmt.from_uint64), (float, pmt.is_real, pmt.to_double, pmt.from_double), (complex, pmt.is_complex, pmt.to_complex, pmt.from_complex), (tuple, pmt.is_tuple, pmt_to_tuple, pmt_from_tuple), (list, pmt.is_vector, pmt_to_vector, pmt_from_vector), (dict, pmt.is_dict, pmt_to_dict, pmt_from_dict), (numpy.ndarray, pmt.is_blob, pmt.blob_data, numpy_to_blob), ) def pmt_to_python(p): for python_type, pmt_check, to_python, from_python in THE_TABLE: if pmt_check(p): return to_python(p) return p #give up, we return the same
raise ValueError("unsupported numpy array dtype for converstion to pmt %s"%(numpy_array.dtype)) def uvector_to_numpy(uvector): match = None for test_func in uvector_mappings.keys(): if test_func(uvector): match = uvector_mappings[test_func] return numpy.array(match[0](uvector), dtype = match[1]) else: raise ValueError("unsupported uvector data type for conversion to numpy array %s"%(uvector)) type_mappings = ( #python type, check pmt type, to python, from python (None, pmt.is_null, lambda x: None, lambda x: pmt.PMT_NIL), (bool, pmt.is_bool, pmt.to_bool, pmt.from_bool), (str, pmt.is_symbol, pmt.symbol_to_string, pmt.string_to_symbol), (unicode, lambda x: False, None, lambda x: pmt.string_to_symbol(x.encode('utf-8'))), (int, pmt.is_integer, pmt.to_long, pmt.from_long), (long, pmt.is_uint64, lambda x: long(pmt.to_uint64(x)), pmt.from_uint64), (float, pmt.is_real, pmt.to_double, pmt.from_double), (complex, pmt.is_complex, pmt.to_complex, pmt.from_complex), (tuple, pmt.is_tuple, pmt_to_tuple, pmt_from_tuple), (list, pmt.is_vector, pmt_to_vector, pmt_from_vector), (dict, pmt.is_dict, pmt_to_dict, pmt_from_dict), (numpy.ndarray, pmt.is_uniform_vector, uvector_to_numpy, numpy_to_uvector), ) def pmt_to_python(p): for python_type, pmt_check, to_python, from_python in type_mappings: if pmt_check(p): return to_python(p) raise ValueError("can't convert %s type to pmt (%s)"%(type(p),p))
raise ValueError("unsupported numpy array dtype for converstion to pmt %s"%(numpy_array.dtype)) def uvector_to_numpy(uvector): match = None for test_func in uvector_mappings.keys(): if test_func(uvector): match = uvector_mappings[test_func] return numpy.array(match[0](uvector), dtype = match[1]) else: raise ValueError("unsupported uvector data type for conversion to numpy array %s"%(uvector)) type_mappings = ( #python type, check pmt type, to python, from python (None, pmt.is_null, lambda x: None, lambda x: PMT_NIL), (bool, pmt.is_bool, pmt.to_bool, pmt.from_bool), (str, pmt.is_symbol, pmt.symbol_to_string, pmt.string_to_symbol), (unicode, lambda x: False, None, lambda x: pmt.string_to_symbol(x.encode('utf-8'))), (int, pmt.is_integer, pmt.to_long, pmt.from_long), (long, pmt.is_uint64, lambda x: long(pmt.to_uint64(x)), pmt.from_uint64), (float, pmt.is_real, pmt.to_double, pmt.from_double), (complex, pmt.is_complex, pmt.to_complex, pmt.from_complex), (tuple, pmt.is_tuple, pmt_to_tuple, pmt_from_tuple), (list, pmt.is_vector, pmt_to_vector, pmt_from_vector), (dict, pmt.is_dict, pmt_to_dict, pmt_from_dict), (tuple, pmt.is_pair, lambda x: (pmt_to_python(pmt.car(x)), pmt_to_python(pmt.cdr(x))), lambda x: pmt.cons(python_to_pmt(x[0]), python_to_pmt(x[1]))), (numpy.ndarray, pmt.is_uniform_vector, uvector_to_numpy, numpy_to_uvector), ) def pmt_to_python(p): for python_type, pmt_check, to_python, from_python in type_mappings: if pmt_check(p): try: