code: def test_set_dict_keys():
      d = {"a": 1, "b": 2, "c": 3}
      assert list(
          py_datastructures.set_dict_keys(d, ["a", "b", "c"])
      ) == ["a", "b", "c"]
      assert list(
          py_datastructures.set_dict_keys(d, ["a", "c", "b"])
      ) == ["a", "c", "b"]
      with pytest.raises(AssertionError):
          py_datastructures.set_dict_keys(d, ["a", "b"])
  
file_path: jiant/tests/utils/python/test_datastructures.py
imports: import json
  import pytest
  from dataclasses import dataclass
  import jiant.utils.python.datastructures
label_desc: Use image node_id 11 for calling a global function with example usage: test_set_dict_keys() without return types