Fixing cli test failure in CI (#44679)

* Fixing cli test failure in CI * review comments (cherry picked from commit 98e0977)
apache · Dec 24, 2024 · 17aa106 · 17aa106
1 parent c95b84c
commit 17aa106
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/cli/test_cli_parser.py b/tests/cli/test_cli_parser.py
@@ -418,10 +418,8 @@ def test_invalid_choice_raises_for_export_format_in_db_export_archived_command(
                     ["db", "export-archived", "--export-format", export_format, "--output-path", "mydir"]
                 )
             error_msg = stderr.getvalue()
-        assert error_msg == (
-            "\nairflow db export-archived command error: argument "
-            f"--export-format: invalid choice: '{export_format}' "
-            "(choose from 'csv'), see help above.\n"
+        assert (
+            "airflow db export-archived command error: argument --export-format: invalid choice" in error_msg
         )
 
     @pytest.mark.parametrize(