def test_packing(self):
        """Ensures the list of lists of tensors gets packed correctly."""
        for batch_constraints, expected_tensor in self.examples:
            packed = pack_constraints(batch_constraints)
            assert torch.equal(packed, expected_tensor)
Ensures the list of lists of tensors gets packed correctly.
decoding/IAD/fairseq/tests/test_constraints.py