code: def test_random_initialization_stratified(
      self, stratified_sampling_mock
  ):
      n_samples = 10
      y = np.array([0] * 10 + [1] * 10 + [2] * 10 + [3] * 70)
      indices = random_initialization_stratified(y, n_samples=n_samples)
      stratified_labels = [y[i] for i in indices]
  
      stratified_sampling_mock.assert_called_with(
          y, n_samples=n_samples
      )
      self.assertEqual(10, len(indices))
      self.assertEqual(10, len(np.unique(indices)))
      self.assertEqual(4, len(np.unique(stratified_labels)))
  
imports: import unittest
  import numpy
  from unittest.mock import patch
  from scipy.sparse import csr_matrix
  from small_text.data import balanced_sampling, stratified_sampling
  from small_text.initialization import random_initialization, random_initialization_stratified, random_initialization_balanced
  from tests.utils.datasets import random_sklearn_dataset
label_desc: Use image node_id 1 for calling the RandomInitializationStratifiedTest obj's underlying member method code with example usage: obj.test_random_initialization_stratified(stratified_sampling_mock) without return types