code: def step(self, action):
      self.state += 1
      done = self.state >= 6
      reward = float(self.state) / 10.0
      return np.array(self.state), reward, done, {}
  
imports: from __future__ import division
  import pytest
  import numpy
  from numpy.testing import assert_allclose
  from rl.memory import SequentialMemory
  from rl.core import Agent, Env, Processor
label_desc: Use image node_id 2 for calling the TestEnv obj's underlying member method code with example usage: obj.step(action) and returns: np,  reward,  done,  dict