code: def test_context_iterator(self, _):
      # Given
      dataset = {"intents": {"dummy": {"utterances": list(range(3))}}}
      random_state = np.random.RandomState(1)
  
      # When
      it = get_contexts_iterator(dataset, "dummy", random_state)
      context = [next(it) for _ in range(5)]
  
      # Then
      self.assertEqual(context, [0, 2, 1, 0, 2])
  
file_path: snips-nlu/snips_nlu/tests/test_data_augmentation.py
imports: from __future__ import unicode_literals
  from builtins import next, range
  import numpy
  from mock import patch
  from snips_nlu.constants import LANGUAGE_EN, STOP_WORDS
  from snips_nlu.data_augmentation import capitalize, capitalize_utterances, generate_utterance, get_contexts_iterator, get_entities_iterators
  from snips_nlu.tests.utils import SnipsTest
label_desc: Use image node_id 1 for calling the TestDataAugmentation obj's underlying member method code with example usage: obj.test_context_iterator(_) without return types