code: def test_gaussian_basic():
      # Data is just 6 separable points in the plane
      X = cp.array(
          [
              [-2, -1, -1],
              [-1, -1, -1],
              [-1, -2, -1],
              [1, 1, 1],
              [1, 2, 1],
              [2, 1, 1],
          ],
          dtype=cp.float32,
      )
      y = cp.array([1, 1, 1, 2, 2, 2])
  
      skclf = skGNB()
      skclf.fit(X.get(), y.get())
  
      clf = GaussianNB()
      clf.fit(X, y)
  
      assert_array_almost_equal(clf.theta_.get(), skclf.theta_, 6)
      assert_array_almost_equal(clf.sigma_.get(), skclf.var_, 6)
  
      y_pred = clf.predict(X)
      y_pred_proba = clf.predict_proba(X)
      y_pred_log_proba = clf.predict_log_proba(X)
      y_pred_proba_sk = skclf.predict_proba(X.get())
      y_pred_log_proba_sk = skclf.predict_log_proba(X.get())
  
      assert_array_equal(y_pred.get(), y.get())
      assert_array_almost_equal(y_pred_proba.get(), y_pred_proba_sk, 8)
      assert_allclose(
          y_pred_log_proba.get(),
          y_pred_log_proba_sk,
          atol=1e-2,
          rtol=1e-2,
      )
  
imports: from cuml.internals.safe_imports import cpu_only_import
  import math
  from sklearn.naive_bayes import GaussianNB
  from sklearn.naive_bayes import ComplementNB
  from sklearn.naive_bayes import CategoricalNB
  from sklearn.naive_bayes import BernoulliNB
  from sklearn.naive_bayes import MultinomialNB
  from numpy.testing import assert_array_almost_equal, assert_raises
  from numpy.testing import assert_allclose, assert_array_equal
  from cuml.datasets import make_classification
  from cuml.internals.input_utils import sparse_scipy_to_cp
  from cuml.naive_bayes import GaussianNB
  from cuml.naive_bayes import ComplementNB
  from cuml.naive_bayes import CategoricalNB
  from cuml.naive_bayes import BernoulliNB
  from cuml.naive_bayes import MultinomialNB
  from sklearn.metrics import accuracy_score
  import pytest
  from cuml.internals.safe_imports import gpu_only_import
label_desc: Use image node_id 9 for calling a global function with example usage: test_gaussian_basic() without return types