code: def test_estimator_checks(test_fn):
      regr_min_zero = ProbWeightRegression(non_negative=True)
      test_fn(
          ProbWeightRegression.__name__ + "_min_zero_true",
          regr_min_zero,
      )
      regr_not_min_zero = ProbWeightRegression(non_negative=False)
      test_fn(
          ProbWeightRegression.__name__ + "_min_zero_true_false",
          regr_not_min_zero,
      )
  
imports: import numpy
  import pytest
  from sklego.common import flatten
  from sklego.linear_model import ProbWeightRegression
  from tests.conftest import general_checks, nonmeta_checks, regressor_checks, select_tests
label_desc: Use image node_id 1 for calling a global function with example usage: test_estimator_checks(test_fn) without return types