code: def test_ni_sit_multi_dataset_merge(self):
      split_mapping = [0, 1, 2, 3, 4, 0, 1, 2, 3, 4]
      mnist_train = MNIST(
          root=default_dataset_location("mnist"),
          train=True,
          download=True,
      )
      mnist_test = MNIST(
          root=default_dataset_location("mnist"),
          train=False,
          download=True,
      )
  
      train_part1 = make_nc_transformation_subset(
          mnist_train, None, None, range(5)
      )
      train_part2 = make_nc_transformation_subset(
          mnist_train, None, None, range(5, 10)
      )
      train_part2 = _taskaware_classification_subset(
          train_part2, class_mapping=split_mapping
      )
  
      test_part1 = make_nc_transformation_subset(
          mnist_test, None, None, range(5)
      )
      test_part2 = make_nc_transformation_subset(
          mnist_test, None, None, range(5, 10)
      )
      test_part2 = _taskaware_classification_subset(
          test_part2, class_mapping=split_mapping
      )
      my_ni_benchmark = ni_benchmark(
          [train_part1, train_part2],
          [test_part1, test_part2],
          5,
          shuffle=True,
          seed=1234,
          balance_experiences=True,
      )
  
      self.assertEqual(5, my_ni_benchmark.n_experiences)
      self.assertEqual(10, my_ni_benchmark.n_classes)
      for batch_id in range(5):
          self.assertEqual(
              10,
              len(
                  my_ni_benchmark.classes_in_experience["train"][
                      batch_id
                  ]
              ),
          )
  
      all_classes = set()
      for batch_id in range(5):
          all_classes.update(
              my_ni_benchmark.classes_in_experience["train"][batch_id]
          )
  
      self.assertEqual(10, len(all_classes))
  
file_path: avalanche/tests/benchmarks/scenarios/deprecated/test_ni_sit_scenario.py
imports: import unittest
  import torch
  from torchvision.datasets import MNIST
  from avalanche.benchmarks import ni_benchmark
  from avalanche.benchmarks.datasets import default_dataset_location
  from avalanche.benchmarks.scenarios import NIExperience, ClassificationStream
  from avalanche.benchmarks.scenarios.deprecated.new_classes.nc_utils import make_nc_transformation_subset
  from avalanche.benchmarks.utils import _taskaware_classification_subset
label_desc: Use image node_id 4 for calling the NISITTests obj's underlying member method code with example usage: obj.test_ni_sit_multi_dataset_merge() without return types