code: def test_when_evaluate_marginal_expectation_without_averaging_result_then_returned_results_have_correct_format():
      # Just checking formats, i.e. no need for correlation.
      X = np.random.normal(0, 1, (1000, 3))
      Y = np.random.normal(0, 1, (1000, 1))
  
      model_all_features = create_linear_regressor()
      model_all_features.fit(X, Y)
  
      results = marginal_expectation(
          model_all_features.predict,
          X,
          X,
          [0],
          return_averaged_results=False,
      )
      assert results.shape[0] == 1000
      assert results.shape[1] == 1000
      assert results.shape[2] == 1
  
imports: import numpy
  import pytest
  from flaky import flaky
  from numpy.matlib import repmat
  from pytest import approx
  from dowhy.gcm.ml import create_hist_gradient_boost_classifier, create_hist_gradient_boost_regressor, create_linear_regressor, create_logistic_regression_classifier
  from dowhy.gcm.stats import estimate_ftest_pvalue, marginal_expectation, quantile_based_fwer
  from dowhy.gcm.util.general import geometric_median
label_desc: Use image node_id 6 for calling a global function with example usage: test_when_evaluate_marginal_expectation_without_averaging_result_then_returned_results_have_correct_format() without return types