Merge pull request #899 from slowder:slowder_fix_test_assertion

PiperOrigin-RevId: 591140634
Change-Id: I1b63d8122b488fb8e730087190b51ba498132392

tf_agents/environments/wrappers_test.py

@@ -289,7 +289,7 @@ def test_fixed_length(self):
         time_step = env.step(np.array(1, dtype=np.int32))
         num_steps += 1
       # Verify episode length.
-      self.assertTrue(num_steps, env.fix_length)
+      self.assertEqual(num_steps, env.fix_length + 1)
 
       # it should automatically reset.
       self.assertTrue(time_step.is_first())