code: def test_plot_model_return_data_original_data(data, plot):
      """Tests whether the return_data parameter of the plot_model function works
      properly or not for the original data
      """
      exp = TSForecastingExperiment()
  
      fh = np.arange(1, 13)
      fold = 2
  
      exp.setup(
          data=data,
          fh=fh,
          fold=fold,
          fold_strategy="sliding",
          verbose=False,
          session_id=42,
      )
  
      plot_data = exp.plot_model(plot=plot, return_data=True)
      # If plot is successful, it will return a dictionary
      # If plot is not possible (e.g. decomposition without index), then it will return None
      assert isinstance(plot_data, dict) or plot_data is None
  
file_path: pycaret/tests/test_time_series_plots.py
imports: import os
  import sys
  import numpy
  import pytest
  from time_series_test_utils import _ALL_PLOTS_DATA, _ALL_PLOTS_ESTIMATOR, _ALL_PLOTS_ESTIMATOR_NOT_DATA, _return_all_plots_estimator_ts_results, _return_data_with_without_period_index, _return_model_names_for_plots_stats
  from pycaret.datasets import get_data
  from pycaret.time_series import TSForecastingExperiment
label_desc: Use image node_id 5 for calling a global function with example usage: test_plot_model_return_data_original_data(data, plot) without return types