code: def test_smoke():
      a = nb.GaussianNB()
      b = nb_.GaussianNB()
      a.fit(X, y)
      X_ = X.compute()
      y_ = y.compute()
      b.fit(X_, y_)
  
      assert_eq(a.class_prior_.compute(), b.class_prior_)
      assert_eq(a.class_count_.compute(), b.class_count_)
      assert_eq(a.theta_.compute(), b.theta_)
  
      assert_eq(a.predict_proba(X).compute(), b.predict_proba(X_))
      assert_eq(a.predict(X).compute(), b.predict(X_))
      assert_eq(
          a.predict_log_proba(X).compute(), b.predict_log_proba(X_)
      )
  
imports: from dask.array.utils import assert_eq
  from sklearn import naive_bayes
  from dask_ml import naive_bayes
  from dask_ml.datasets import make_classification
label_desc: Use image node_id 1 for calling a global function with example usage: test_smoke() without return types