def METHOD_NAME(test_input, expected):
    assert torch.all(
        torch.eq(
            spikegen.targets_convert(
                test_input,
                num_classes=5,
                code="rate",
                num_steps=5,
                first_spike_time=2,
            )[:, 0, 4],
            expected,
        )
    )