def METHOD_NAME(self):
    unprocessed_tokens = torch.tensor([2, 2, 0, 3, 3, 3, 0, 3])
    decoder = self._get_decoder()
    timesteps = decoder._get_timesteps(unprocessed_tokens)
    expected = [0, 3, 7]
    self.assertEqual(timesteps, expected)