def test_replace_single_value_cpu(): template.replace_single_value(SINGLE_VALUES_SET, new_value="5", current_value="20", key="worker_cpu") assert SINGLE_VALUES_SET.get("worker_cpu") == "2.5"
def test_replace_single_value_empty(): template.replace_single_value(SINGLE_VALUES_SET, new_value="10G", current_value="25G", key="memory", cpu=False) assert SINGLE_VALUES_SET.get("memory") == "null"
def test_replace_single_value_cpu_for_horovod(): template.replace_single_value(SINGLE_VALUES_SET, new_value="5", current_value="20", key="cpus", round_to_int=True) assert SINGLE_VALUES_SET.get("cpus") == "1"