fix test due to new CLI arg description handling

ecmwf · Mar 14, 2024 · 3606267 · 3606267
1 parent d95ae92
commit 3606267
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_conflator_functions.py b/tests/test_conflator_functions.py
@@ -41,7 +41,7 @@ class Config(ConfigModel):
             assert len(cli_args) == 1
             for arg in cli_args:
                 assert arg.argparse_key == "test_email"
-                assert arg.description is None
+                assert arg.description == ""
                 assert arg.args == ("--test-email",)
 
         class NewConfig(ConfigModel):
@@ -58,7 +58,7 @@ class NewConfig(ConfigModel):
             assert len(cli_args) == 2
             for arg in cli_args:
                 assert arg.argparse_key in ["test_email", "test_key"]
-                assert arg.description is None
+                assert arg.description in ["Test API Key", ""]
                 if arg.argparse_key == "test_email":
                     assert arg.args == ("--test-email",)
                 else: