def testStochasticTwoFrames(self):
     self.TestOnVariousInputOutputSizes(
         next_frame_params.next_frame_stochastic(),
         next_frame_sv2p.NextFrameStochasticTwoFrames, 1)
 def testStochasticWithActionsAndRewards(self):
     self.TestWithActionAndRewards(
         next_frame_params.next_frame_stochastic(),
         next_frame_sv2p.NextFrameStochastic, 1)