diff --git a/leetcode/accounts_merge/test_solution.py b/leetcode/accounts_merge/test_solution.py index c6244883..f8341af9 100644 --- a/leetcode/accounts_merge/test_solution.py +++ b/leetcode/accounts_merge/test_solution.py @@ -58,8 +58,8 @@ def setup_method(self): ( [ ["A", "a@mail.com", "b@mail.com"], - ["B", "b@mail.com", "c@mail.com"], - ["C", "c@mail.com", "d@mail.com"], + ["A", "b@mail.com", "c@mail.com"], + ["A", "c@mail.com", "d@mail.com"], ], [["A", "a@mail.com", "b@mail.com", "c@mail.com", "d@mail.com"]], ), diff --git a/leetcode/two_sum/helpers.py b/leetcode/two_sum/helpers.py index 00f7b734..2876b5cf 100644 --- a/leetcode/two_sum/helpers.py +++ b/leetcode/two_sum/helpers.py @@ -4,5 +4,8 @@ def run_two_sum(solution_class: type, nums: list[int], target: int): def assert_two_sum(result: list[int], expected: list[int]) -> bool: - assert result == expected + # Sort both result and expected for comparison since order doesn't matter + result_sorted = sorted(result) + expected_sorted = sorted(expected) + assert result_sorted == expected_sorted return True diff --git a/leetcode/two_sum/test_solution.py b/leetcode/two_sum/test_solution.py index aefbad0f..de814b7d 100644 --- a/leetcode/two_sum/test_solution.py +++ b/leetcode/two_sum/test_solution.py @@ -36,4 +36,4 @@ def setup_method(self): ) def test_two_sum(self, nums: list[int], target: int, expected: list[int]): result = run_two_sum(Solution, nums, target) - assert_two_sum(sorted(result), sorted(expected)) + assert_two_sum(result, expected)