Added contains_keys assert function