def setup_for_test(cls):
    """Setup directories and files required to run the problem."""

    tmp_dir = tf.test.get_temp_dir()
    shutil.rmtree(tmp_dir)
    os.mkdir(tmp_dir)
    cls.data_dir = tmp_dir

    # Generate a small test dataset
    cls().generate_data(TinyAlgo.data_dir, None)
Setup directories and files required to run the problem.
tensor2tensor/data_generators/algorithmic.py