code: def test_sad(max_sample, expected):
      dataset = ActiveLearningDataset(
          MyDataset(), make_unlabelled=lambda x: -1
      )
      active_loop = ActiveLearningLoop(
          dataset,
          get_probs_iter,
          heuristics.Random(),
          max_sample=max_sample,
          query_size=10,
          dummy_param=1,
      )
      dataset.label_randomly(10)
      active_loop.step()
      assert len(dataset) == 10 + expected
  
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 5 for calling a global function with example usage: test_sad(max_sample, expected) without return types