import tensorflow as tf

    self._testSampleTaskHelper(p)

  def testSampleTask(self):
    """Expected distribution: 'a': 0.8 , 'b': 0.2."""
    p = self._setUpTestSampleTask()

    p.task_schedule = task_scheduler.ConstantScheduler.Params()
    p.task_schedule.task_probs = [('a', 0.8), ('b', 0.2)]

    self._testSampleTaskHelper(p)


if __name__ == '__main__':
  tf.test.main()
