code: def test_stats_models_wrapper():
      """
      A trivial test of the StatsModelsWrapper
      """
      X = np.array([[1], [2], [3]])
      y = np.array([1.1, 2, 3])
  
      glm_gaussian = functools.partial(
          sm.GLM, family=sm.families.Gaussian()
      )
      sm_est = StatsModelsWrapper(glm_gaussian)
  
      assert sm_est.fit(X, y) is sm_est, "fit did not return self"
      assert sm_est.predict(X).shape == (3,)
      assert 0.0 <= sm_est.score(X, y) <= 1.0
  
imports: import pytest
  import functools
  import numpy
  from yellowbrick.contrib.statsmodels import StatsModelsWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: test_stats_models_wrapper() without return types