diff options
Diffstat (limited to 'cli/tests/test_models_settings.py')
-rw-r--r-- | cli/tests/test_models_settings.py | 52 |
1 files changed, 25 insertions, 27 deletions
diff --git a/cli/tests/test_models_settings.py b/cli/tests/test_models_settings.py index 0834355516f..f9f2a936cd1 100644 --- a/cli/tests/test_models_settings.py +++ b/cli/tests/test_models_settings.py @@ -9,32 +9,31 @@ from openbb_cli.models.settings import Settings def test_default_values(): """Test the default values of the settings model.""" - settings = Settings() - assert settings.VERSION == "1.0.0" - assert settings.TEST_MODE is False - assert settings.DEBUG_MODE is False - assert settings.DEV_BACKEND is False - assert settings.FILE_OVERWRITE is False - assert settings.SHOW_VERSION is True - assert settings.USE_INTERACTIVE_DF is True - assert settings.USE_CLEAR_AFTER_CMD is False - assert settings.USE_DATETIME is True - assert settings.USE_PROMPT_TOOLKIT is True - assert settings.ENABLE_EXIT_AUTO_HELP is True - assert settings.REMEMBER_CONTEXTS is True - assert settings.ENABLE_RICH_PANEL is True - assert settings.TOOLBAR_HINT is True - assert settings.SHOW_MSG_OBBJECT_REGISTRY is False - assert settings.TIMEZONE == "America/New_York" - assert settings.FLAIR == ":openbb" - assert settings.PREVIOUS_USE is False - assert settings.N_TO_KEEP_OBBJECT_REGISTRY == 10 - assert settings.N_TO_DISPLAY_OBBJECT_REGISTRY == 5 - assert settings.RICH_STYLE == "dark" - assert settings.ALLOWED_NUMBER_OF_ROWS == 20 - assert settings.ALLOWED_NUMBER_OF_COLUMNS == 5 - assert settings.HUB_URL == "https://my.openbb.co" - assert settings.BASE_URL == "https://payments.openbb.co" + fields = Settings.model_fields + assert fields["TEST_MODE"].default is False + assert fields["DEBUG_MODE"].default is False + assert fields["DEV_BACKEND"].default is False + assert fields["FILE_OVERWRITE"].default is False + assert fields["SHOW_VERSION"].default is True + assert fields["USE_INTERACTIVE_DF"].default is True + assert fields["USE_CLEAR_AFTER_CMD"].default is False + assert fields["USE_DATETIME"].default is True + assert fields["USE_PROMPT_TOOLKIT"].default is True + assert fields["ENABLE_EXIT_AUTO_HELP"].default is True + assert fields["REMEMBER_CONTEXTS"].default is True + assert fields["ENABLE_RICH_PANEL"].default is True + assert fields["TOOLBAR_HINT"].default is True + assert fields["SHOW_MSG_OBBJECT_REGISTRY"].default is False + assert fields["TIMEZONE"].default == "America/New_York" + assert fields["FLAIR"].default == ":openbb" + assert fields["PREVIOUS_USE"].default is False + assert fields["N_TO_KEEP_OBBJECT_REGISTRY"].default == 10 + assert fields["N_TO_DISPLAY_OBBJECT_REGISTRY"].default == 5 + assert fields["RICH_STYLE"].default == "dark" + assert fields["ALLOWED_NUMBER_OF_ROWS"].default == 20 + assert fields["ALLOWED_NUMBER_OF_COLUMNS"].default == 5 + assert fields["HUB_URL"].default == "https://my.openbb.co" + assert fields["BASE_URL"].default == "https://payments.openbb.co" # Test __repr__ output @@ -43,7 +42,6 @@ def test_repr(): settings = Settings() repr_str = settings.__repr__() # pylint: disable=C2801 assert "Settings\n\n" in repr_str - assert "VERSION: 1.0.0" in repr_str # Test loading from environment variables |