def test_add_attributes(key, value, arg):
class TestModelVersion:
def add_attribute(self, key, value0, overwrite):
assert value == value0
def _get_attribute_keys(self):
return [key]
model_version = TestModelVersion()
add_attributes(model_version, arg, True)
def test_multiple_attributes():
call_number = [0]
values = [3.6, {"a": 1, "b": 2}]
class TestModelVersion:
def add_attribute(self, key, value0, overwrite):
assert values[call_number[0]] == value0
call_number[0] += 1
def _get_attribute_keys(self):
return ["numl", "dict"]
model_version = TestModelVersion()
add_attributes(model_version, ["num=3.6", 'dict={"a": 1, "b": 2}'], True)
