def test_multiple_completions(self):
        """Test handling multiple completions at once."""
        completions = [
            [{"content": r"\boxed{\frac{63}{400}}"}],
            [{"content": r"\boxed{\frac{64}{400}}"}],
        ]
        solutions = [r"\frac{63}{400}", r"\frac{63}{400}"]

        rewards = accuracy_reward(completions, solutions)
        self.assertEqual(len(rewards), 2)
        self.assertEqual(rewards[0], 1.0)
        self.assertEqual(rewards[1], 0.0)
Test handling multiple completions at once.
tests/test_rewards.py