def test_add_get_cpp_types_with_datum():
from kwiver.sprokit.adapters import adapter_data_set
ads = adapter_data_set.AdapterDataSet.create()
add_get_helper(
ads,
ads.add_datum,
ads._get_port_data_double_vector,
datum.new_double_vector(datum.VectorDouble([6.3, 8.9])),
"datum_double_vector",
)
add_get_helper(
ads,
ads.add_datum,
ads._get_port_data_string_vector,
datum.new_string_vector(datum.VectorString(["foo", "bar"])),
"datum_string_vector",
)
add_get_helper(
ads,
ads.add_datum,
ads._get_port_data_uchar_vector,
datum.new_uchar_vector(datum.VectorUChar([102, 103])),
"datum_uchar_vector",
)
def test_mix_new_and_get():
from kwiver.sprokit.pipeline import datum
from kwiver.vital import types as kvt
# Try creating with generic constructor first, retrieving with
# type specific get function
datum_inst = datum.new("string_value")
check_same_type(datum_inst.get_string(), "string_value")
datum_inst = datum.new(kvt.Timestamp(1000000000, 10))
check_same_type(datum_inst.get_timestamp(), kvt.Timestamp(1000000000, 10))
datum_inst = datum.new(datum.VectorString(["element1", "element2"]))
check_same_type(datum_inst.get_string_vector(),
datum.VectorString(["element1", "element2"]))
# Now try the opposite
datum_inst = datum.new_string("string_value")
check_same_type(datum_inst.get_datum(), "string_value")
datum_inst = datum.new_timestamp(kvt.Timestamp(1000000000, 10))
check_same_type(datum_inst.get_datum(), kvt.Timestamp(1000000000, 10))
datum_inst = datum.new_string_vector(
datum.VectorString(["element1", "element2"]))
check_same_type(datum_inst.get_datum(),
datum.VectorString(["element1", "element2"]))
def test_add_get_cpp_types():
from kwiver.sprokit.pipeline import datum
datum_inst = datum.new_double_vector(datum.VectorDouble([3.14, 4.14]))
retrieved_val = datum_inst.get_double_vector()
check_same_type(retrieved_val, datum.VectorDouble([3.14, 4.14]))
datum_inst = datum.new_string_vector(datum.VectorString(["s00", "s01"]))
retrieved_val = datum_inst.get_string_vector()
check_same_type(retrieved_val, datum.VectorString(["s00", "s01"]))
datum_inst = datum.new_uchar_vector(datum.VectorUChar([100, 101]))
retrieved_val = datum_inst.get_uchar_vector()
check_same_type(retrieved_val, datum.VectorUChar([100, 101]))
check_automatic_conversion(datum.VectorDouble([3.14, 4.14]))
check_automatic_conversion(datum.VectorString(["s00", "s01"]))
check_automatic_conversion(datum.VectorUChar([100, 101]))