code: def test_simple_query(self, num_classes=3, num_samples=20):
      clf = self._get_clf()
      dataset = random_sklearn_dataset(
          num_samples, vocab_size=10, num_classes=num_classes
      )
  
      strategy = self._get_query_strategy()
      indices_labeled = np.random.choice(
          np.arange(num_samples), size=10, replace=False
      )
      indices_unlabeled = np.array(
          [
              i
              for i in range(len(dataset))
              if i not in set(indices_labeled)
          ]
      )
  
      with self.assertRaisesRegex(
          TypeError, r"predict_proba\(\) got an unexpected keyword arg"
      ):
          strategy.query(
              clf,
              dataset,
              indices_unlabeled,
              indices_labeled,
              self._get_query_strategy(),
              n=5,
          )
  
file_path: small-text/tests/unit/small_text/query_strategies/test_bayesian.py
imports: import unittest
  import numpy
  from small_text.classifiers import ConfidenceEnhancedLinearSVC
  from small_text.query_strategies.bayesian import BALD, _bald
  from tests.utils.datasets import random_sklearn_dataset
  from tests.utils.testing import assert_array_equal
label_desc: Use image node_id 3 for calling the BALDTest obj's underlying member method code with example usage: obj.test_simple_query(num_classes, num_samples) without return types