def testSingleElem(self):
    """Tests Retrace with a single element batch and lambda set to 1.0."""
    batch_size = 1
    lambda_ = 1.0
    self._main_test(batch_size, lambda_)
Tests Retrace with a single element batch and lambda set to 1.0.
trfl/continuous_retrace_ops_test.py