From 6758bed4a5ebf5bf2649ae510a8b0e67f317a07b Mon Sep 17 00:00:00 2001 From: Tom Ward Date: Fri, 19 Jun 2020 05:43:41 -0700 Subject: [PATCH] Replace self.assertEqual(None,...) with self.assetIsNone(...) PiperOrigin-RevId: 317292604 Change-Id: I49521b4b52d4fc587de8f48395d2931f9f9dc2cb --- dm_env_rpc/v1/dm_env_adaptor_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dm_env_rpc/v1/dm_env_adaptor_test.py b/dm_env_rpc/v1/dm_env_adaptor_test.py index 10e8688..3ef151e 100644 --- a/dm_env_rpc/v1/dm_env_adaptor_test.py +++ b/dm_env_rpc/v1/dm_env_adaptor_test.py @@ -152,8 +152,8 @@ def test_first_running_step(self): self._connection.send.assert_called_once_with(_SAMPLE_STEP_REQUEST) self.assertEqual(dm_env.StepType.FIRST, timestep.step_type) - self.assertEqual(None, timestep.reward) - self.assertEqual(None, timestep.discount) + self.assertIsNone(timestep.reward) + self.assertIsNone(timestep.discount) self.assertEqual({'foo': 5, 'bar': 'goodbye'}, timestep.observation) def test_mid_running_step(self): @@ -195,8 +195,8 @@ def test_reset(self): timestep = self._env.reset() self.assertEqual(dm_env.StepType.FIRST, timestep.step_type) - self.assertEqual(None, timestep.reward) - self.assertEqual(None, timestep.discount) + self.assertIsNone(timestep.reward) + self.assertIsNone(timestep.discount) self.assertEqual({'foo': 5, 'bar': 'goodbye'}, timestep.observation) def test_reset_changes_spec_raises_error(self):