diff --git a/interface_tester/interface_test.py b/interface_tester/interface_test.py index 86144c2..4c013a1 100644 --- a/interface_tester/interface_test.py +++ b/interface_tester/interface_test.py @@ -297,7 +297,7 @@ def assert_schema_valid(self, schema: Optional["DataBagSchema"] = None): { "unit": relation.local_unit_data, "app": relation.local_app_data, - } + }, ) except ValidationError as e: errors.append(str(e)) @@ -451,7 +451,7 @@ def _get_endpoint(supported_endpoints: dict, role: Role, interface_name: str): return endpoints_for_interface[0] def _generate_relations_state( - self, state_template: State, input_state: State, supported_endpoints, role: Role + self, state_template: State, input_state: State, supported_endpoints, role: Role ) -> List[Relation]: """Merge the relations from the input state and the state template into one. diff --git a/tests/unit/test_collect_schemas.py b/tests/unit/test_collect_schemas.py index 5e97a1d..7357da4 100644 --- a/tests/unit/test_collect_schemas.py +++ b/tests/unit/test_collect_schemas.py @@ -3,12 +3,12 @@ import pytest -from interface_tester.interface_test import _has_pydantic_v1 from interface_tester.collector import ( collect_tests, get_schema_from_module, load_schema_module, ) +from interface_tester.interface_test import _has_pydantic_v1 def test_load_schema_module(tmp_path): @@ -76,8 +76,12 @@ class RequirerSchema(DataBagSchema): assert tests["mytestinterfaceb"]["v0"]["requirer"]["schema"].__fields__["foo"].default == 2 else: - assert tests["mytestinterfacea"]["v0"]["requirer"]["schema"].model_fields["foo"].default == 1 - assert tests["mytestinterfaceb"]["v0"]["requirer"]["schema"].model_fields["foo"].default == 2 + assert ( + tests["mytestinterfacea"]["v0"]["requirer"]["schema"].model_fields["foo"].default == 1 + ) + assert ( + tests["mytestinterfaceb"]["v0"]["requirer"]["schema"].model_fields["foo"].default == 2 + ) def test_collect_invalid_schemas(tmp_path):