def test_component_set_field_with_scalar_value_correctly_sets_field_value(): comp = Component("comp4") field_name = "testfield" data = 123 dtype = ValueTypes.INT comp.set_field_value(field_name, data, dtype) field_dataset = comp[field_name] assert field_dataset.name == field_name assert field_dataset.values == data assert field_dataset.type == dtype
def test_component_set_field_with_numpy_array_correctly_sets_field_value(): comp = Component("comp4") data = [[1], [2]] dtype = ValueTypes.INT field_name = "field1" field_value = np.asarray(data, dtype=int) comp.set_field_value(field_name, field_value, dtype) field_dataset = comp["field1"] assert field_dataset.name == field_name assert np.array_equal(field_dataset.values, field_value) assert field_dataset.type == dtype
def _add_component_to_file(field_name: str, field_value: Any, component_name: str = "test_component"): component = Component(component_name) component.set_field_value( field_name, Dataset( parent_node=None, name=field_name, type=ValueTypes.DOUBLE, size="[1]", values=field_value, ), dtype=ValueTypes.DOUBLE, ) return component