def METHOD_NAME(test_input, expected):
    assert torch.all(
        torch.eq(
            spikegen.to_one_hot_inverse(
                spikegen.to_one_hot(test_input.squeeze(0), 2)
            ),
            expected,
        )
    )