def test_clear_input_files(context_with_data: CodeExecutorContext):
  """Test clearing input files and processed file names."""
  context_with_data.clear_input_files()
  assert context_with_data._session_state["_code_executor_input_files"] == []
  assert context_with_data._context["processed_input_files"] == []
Test clearing input files and processed file names.
tests/unittests/code_executors/test_code_executor_context.py