class_docstr: Base helper class for next frame tests.
code: def TestWithActionAndRewards(
      self, hparams, model, expected_last_dim, test_infer=True
  ):
      test_funcs = [self.TestVideoModelWithActionAndRewards]
      if test_infer:
          test_funcs += [self.TestVideoModelWithActionAndRewardsInfer]
      for test_func in test_funcs:
          test_func(1, 1, hparams, model, expected_last_dim)
          test_func(1, 6, hparams, model, expected_last_dim)
          test_func(4, 1, hparams, model, expected_last_dim)
          test_func(7, 5, hparams, model, expected_last_dim)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.data_generators import video_generated
  from tensor2tensor.layers import modalities
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 11 for calling the BaseNextFrameTest obj's underlying member method code with example usage: obj.TestWithActionAndRewards(hparams, model, expected_last_dim, test_infer) without return types