class_docstr: This class implements the base class for all unit tests.
class_docstr_tok: ["This","class","implements","the","base","class","for","all","unit","tests","."]
code: def create_image_dataset(cls, n_train, n_test):
      (
          (x_train_mnist, y_train_mnist),
          (x_test_mnist, y_test_mnist),
          _,
          _,
      ) = load_dataset("mnist")
      cls.x_train_mnist = x_train_mnist[:n_train]
      cls.y_train_mnist = y_train_mnist[:n_train]
      cls.x_test_mnist = x_test_mnist[:n_test]
      cls.y_test_mnist = y_test_mnist[:n_test]
  
      cls._x_train_mnist_original = cls.x_train_mnist.copy()
      cls._y_train_mnist_original = cls.y_train_mnist.copy()
      cls._x_test_mnist_original = cls.x_test_mnist.copy()
      cls._y_test_mnist_original = cls.y_test_mnist.copy()
  
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  import json
  import logging
  import os
  import pickle
  import time
  import unittest
  import warnings
  import sklearn
  import numpy
  from art.estimators.classification.tensorflow import TensorFlowV2Classifier
  from art.estimators.encoding.tensorflow import TensorFlowEncoder
  from art.estimators.generation.tensorflow import TensorFlowGenerator, TensorFlowV2Generator
  from art.estimators.gan.tensorflow import TensorFlowV2GAN
  from art.utils import load_dataset
label_desc: Use image node_id 2 for calling the TestBase obj's underlying member method code with example usage: obj.create_image_dataset(cls, n_train, n_test) without return types