code: def _assert_masks(self, info, control, discount):
      self.assertEqual(
          info, {"control_mask": control, "discount_mask": discount}
      )
  
file_path: trax/trax/rl/envs/data_envs_test.py
imports: from absl.testing import absltest
  import numpy
  from trax.rl.envs import data_envs
label_desc: Use image node_id 1 for calling the SequenceDataEnvTest obj's underlying member method code with example usage: obj._assert_masks(info, control, discount) without return types