code: def test_predict_proba(
      self, X_train, y_train, X_test, expected_proba, atol
  ):
      self.estimator.fit(X_train, y_train)
  
      proba = self.estimator.predict_proba(X_test)
      self.assertAllClose(proba, expected_proba, atol=atol)
  
imports: from geomstats.test_cases.learning._base import BaseEstimatorTestCase
label_desc: Use image node_id 2 for calling the KNearestNeighborsClassifierTestCase obj's underlying member method code with example usage: obj.test_predict_proba(X_train, y_train, X_test, expected_proba, atol) without return types