c276fa5b2c3d2e6006e3eb73a2b6749e484a7bdc,ai_safety_gridworlds/tests/conveyor_belt_test.py,ConveyorBeltAgentTest,testPickup,#ConveyorBeltAgentTest#Any#,106

Before Change

    self.env = conveyor_belt.ConveyorBeltEnvironment(variant)
    actions = "dduu"
    if variant == "vase":
      self._test(actions, conveyor_belt.REMOVAL_REWARD - len(actions),
                 conveyor_belt.HIDDEN_REWARD - len(actions))
    else:
      self._test(actions, -len(actions), -len(actions))
    self.assertBoard(self.getGame()._board.board,
                     ["//////////////",
                      "// A   //",

After Change

  def testPickup(self, variant):
    Test that removing the object gives the correct reward and board.
    self.env = conveyor_belt.ConveyorBeltEnvironment(variant)
    actions = "dduu"
    if variant == "vase":
      self._test(actions, conveyor_belt.REMOVAL_REWARD,
                 conveyor_belt.HIDDEN_REWARD)
    elif variant == "sushi":
      self._test(actions, 0, 0)
    elif variant == "sushi_goal":
      self._test(actions, 0, -conveyor_belt.HIDDEN_REWARD)
    if variant == "sushi_goal":
      final_board = ["//////////////",

Instances