input: Write a function to store and retrieve user data in a JSON file. The function should take a dictionary of user data as input and store it in a JSON file. It should also be able to load and return the user data from the JSON file.
  assert store_user_data({"Name": "John", "Age": 25, "City": "New York"})
  data = get_user_data()
  assert data == {"Name": "John", "Age": 25, "City": "New York"}
output: def store_user_data(user_data):
      with open('user_data.json', 'w') as file:
          json.dump(user_data, file)
      return True
  
  def get_user_data():
      with open('user_data.json', 'r') as file:
          return json.load(file)