code: def test_obvious_usecase(k):
      X = np.concatenate(
          [
              np.random.normal(-10, 1, (100, 2)),
              np.random.normal(10, 1, (100, 2)),
          ]
      )
      y = np.concatenate([np.zeros(100), np.ones(100)])
      assert (
          GaussianMixtureNB(n_components=k, max_iter=1000)
          .fit(X, y)
          .predict(X)
          == y
      ).all()
      assert (
          BayesianGaussianMixtureNB(n_components=k, max_iter=1000)
          .fit(X, y)
          .predict(X)
          == y
      ).all()
  
file_path: scikit-lego/tests/test_estimators/test_gmm_naive_bayes.py
imports: import numpy
  import pytest
  from sklego.common import flatten
  from sklego.naive_bayes import BayesianGaussianMixtureNB, GaussianMixtureNB
  from tests.conftest import general_checks, nonmeta_checks, select_tests
label_desc: Use image node_id 3 for calling a global function with example usage: test_obvious_usecase(k) without return types