Fixing cli test failure in CI (#44679) (#44806)

* Fixing cli test failure in CI * review comments (cherry picked from commit 98e0977) Co-authored-by: Amogh Desai <[email protected]> (cherry picked from commit c942b55)
apache · Dec 10, 2024 · 17495d7 · 17495d7
1 parent 9e3a976
commit 17495d7
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
@@ -417,10 +417,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(