Replace self.assertEqual(None,...) with self.assetIsNone(...)

PiperOrigin-RevId: 317292604 Change-Id: I49521b4b52d4fc587de8f48395d2931f9f9dc2cb
google-deepmind · Jun 22, 2020 · 6758bed · 6758bed
1 parent 9ab3af9
commit 6758bed
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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):