diff --git a/tests/unit/test_schema.py b/tests/unit/test_schema.py index cec1767..cc1aef7 100644 --- a/tests/unit/test_schema.py +++ b/tests/unit/test_schema.py @@ -1,10 +1,15 @@ import pytest + from lighter.utils.runner import parse_config -@pytest.mark.parametrize("config", [ - "./projects/cifar10/experiments/monai_bundle_prototype.yaml", - "./projects/cifar10/experiments/monai_bundle_prototype.yaml,./tests/configs/test1.yaml" -]) + +@pytest.mark.parametrize( + "config", + [ + "./projects/cifar10/experiments/monai_bundle_prototype.yaml", + "./projects/cifar10/experiments/monai_bundle_prototype.yaml,./tests/configs/test1.yaml", + ], +) def test_config_schema_validation(config: str): """ Test the validation of configuration schemas. @@ -16,7 +21,7 @@ def test_config_schema_validation(config: str): config (str): Path to the configuration file(s) to be validated. """ # Parse the configuration file(s) - parsed_config = parse_config(config=config.split(',')) - + parsed_config = parse_config(config=config.split(",")) + # Ensure the parsed configuration is a dictionary assert isinstance(parsed_config.config, dict)