code: def test_should_stop(heur):
      dataset = ActiveLearningDataset(
          MyDataset(), make_unlabelled=lambda x: -1
      )
      active_loop = ActiveLearningLoop(
          dataset, get_probs, heur, query_size=10, dummy_param=1
      )
      dataset.label_randomly(10)
      step = 0
      for _ in range(15):
          flg = active_loop.step()
          step += 1
          if not flg:
              break
  
      assert step == 10
  
imports: import os
  import pickle
  import warnings
  from unittest.mock import patch
  import numpy
  import pytest
  from torch.utils.data import Dataset
  from baal.active import ActiveLearningDataset, heuristics
  from baal.active.active_loop import ActiveLearningLoop
label_desc: Use image node_id 3 for calling a global function with example usage: test_should_stop(heur) without return types