code: def test_when_fitting_bayesian_gaussian_mixture_distribution_then_samples_should_be_drawn_correctly():
      test_data = np.array([[0, 0], [0, 0], [1, 2], [1, 2]])
  
      approximated_data_distribution_model = (
          BayesianGaussianMixtureDistribution()
      )
      approximated_data_distribution_model.fit(test_data)
      assert approximated_data_distribution_model.draw_samples(
          5
      ).shape == (5, 2)
  
file_path: dowhy/tests/gcm/test_stochastic_models.py
imports: import numpy
  import pytest
  from flaky import flaky
  from pytest import approx
  from scipy import stats
  from dowhy.gcm import BayesianGaussianMixtureDistribution, EmpiricalDistribution, ScipyDistribution
label_desc: Use image node_id 1 for calling a global function with example usage: test_when_fitting_bayesian_gaussian_mixture_distribution_then_samples_should_be_drawn_correctly() without return types