code: def test_get_trials_by_state():
      t1 = Trial("__fake", stub=True)
      t1.set_status(Trial.RUNNING)
      t2 = Trial("__fake", stub=True)
      t2.set_status(Trial.PENDING)
      trials = [t1, t2]
      assert _get_trials_by_state(trials) == {
          "RUNNING": [t1],
          "PENDING": [t2],
      }
  
imports: import argparse
  from unittest import mock
  import pytest
  import sys
  from freezegun import freeze_time
  from ray import tune
  from ray.air.constants import TRAINING_ITERATION
  from ray.tune.experimental.output import _get_time_str, _get_trials_by_state, _get_trial_info, _infer_user_metrics, _max_len, _current_best_trial, _best_trial_str, _get_trial_table_data, _get_dict_as_table_data, _infer_params, AirVerbosity, TrainReporter, TuneTerminalReporter
  from ray.tune.experiment.trial import Trial
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_trials_by_state() without return types