human
I wrote the below code

    def check_list_kv_pair(kv, key):
        kv.push(key, ([(mx.nd.ones(shape) * 4)] * len(key)))
        val = ([mx.nd.empty(shape)] * len(key))
        kv.pull(key, out=val)
        for v in val:
            check_diff_to_scalar(v, 4)
    check_list_kv_pair(init_kv(), keys)
    check_list_kv_pair(init_kv_with_str(), str_keys)

What's a good function header?
assistant
#Here's a good function header.

def test_list_kv_pair():