style: apply ruff formatting to test files

Author: mkumar73

tests/test_base.py

@@ -54,9 +54,9 @@ def test_model_has_forward_method(model_class):
     assert hasattr(model_class, "forward"), f"{model_class.__name__} is missing a forward method."
 
     sig = inspect.signature(model_class.forward)
-    assert any(
-        p.kind == inspect.Parameter.VAR_POSITIONAL for p in sig.parameters.values()
-    ), f"{model_class.__name__}.forward should have *data argument."
+    assert any(p.kind == inspect.Parameter.VAR_POSITIONAL for p in sig.parameters.values()), (
+        f"{model_class.__name__}.forward should have *data argument."
+    )
 
 
 @pytest.mark.parametrize("model_class", model_classes)
@@ -77,9 +77,9 @@ def test_model_has_num_classes(model_class):
 def test_model_calls_super_init(model_class):
     """Test that each model calls super().__init__(config=config, **kwargs)."""
     source = inspect.getsource(model_class.__init__)
-    assert (
-        "super().__init__(config=config" in source
-    ), f"{model_class.__name__} should call super().__init__(config=config, **kwargs)."
+    assert "super().__init__(config=config" in source, (
+        f"{model_class.__name__} should call super().__init__(config=config, **kwargs)."
+    )
 
 
 @pytest.mark.parametrize("model_class", model_classes)

tests/test_configs.py

@@ -66,20 +66,20 @@ def test_config_default_values(config_class):
         if origin is typing.Literal:
             # If the field is a Literal, ensure the value is one of the allowed options
             allowed_values = typing.get_args(expected_type)
-            assert (
-                value in allowed_values
-            ), f"{config_class.__name__}.{attr} has incorrect value: expected one of {allowed_values}, got {value}"
+            assert value in allowed_values, (
+                f"{config_class.__name__}.{attr} has incorrect value: expected one of {allowed_values}, got {value}"
+            )
         elif origin is typing.Union:
             # For Union types (e.g., Optional[str]), check if value matches any type in the union
             allowed_types = typing.get_args(expected_type)
-            assert any(
-                isinstance(value, t) for t in allowed_types
-            ), f"{config_class.__name__}.{attr} has incorrect type: expected one of {allowed_types}, got {type(value)}"
+            assert any(isinstance(value, t) for t in allowed_types), (
+                f"{config_class.__name__}.{attr} has incorrect type: expected one of {allowed_types}, got {type(value)}"
+            )
         elif origin is not None:
             # If it's another generic type (e.g., list[str]), check against the base type
-            assert (
-                isinstance(value, origin) or value is None
-            ), f"{config_class.__name__}.{attr} has incorrect type: expected {expected_type}, got {type(value)}"
+            assert isinstance(value, origin) or value is None, (
+                f"{config_class.__name__}.{attr} has incorrect type: expected {expected_type}, got {type(value)}"
+            )
         else:
             # Standard type check
             assert (