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)