rename TestWeights to appease pytest (#5054)

Author: pmeier

test/test_prototype_models.py

@@ -203,27 +203,27 @@ def test_smoke():
 # With this filter, every unexpected warning will be turned into an error
 @pytest.mark.filterwarnings("error")
 class TestHandleLegacyInterface:
-    class TestWeights(WeightsEnum):
+    class ModelWeights(WeightsEnum):
         Sentinel = Weights(url="https://pytorch.org", transforms=lambda x: x, meta=dict())
 
     @pytest.mark.parametrize(
         "kwargs",
         [
             pytest.param(dict(), id="empty"),
             pytest.param(dict(weights=None), id="None"),
-            pytest.param(dict(weights=TestWeights.Sentinel), id="Weights"),
+            pytest.param(dict(weights=ModelWeights.Sentinel), id="Weights"),
         ],
     )
     def test_no_warn(self, kwargs):
-        @handle_legacy_interface(weights=("pretrained", self.TestWeights.Sentinel))
+        @handle_legacy_interface(weights=("pretrained", self.ModelWeights.Sentinel))
         def builder(*, weights=None):
             pass
 
         builder(**kwargs)
 
     @pytest.mark.parametrize("pretrained", (True, False))
     def test_pretrained_pos(self, pretrained):
-        @handle_legacy_interface(weights=("pretrained", self.TestWeights.Sentinel))
+        @handle_legacy_interface(weights=("pretrained", self.ModelWeights.Sentinel))
         def builder(*, weights=None):
             pass
 
@@ -232,7 +232,7 @@ def builder(*, weights=None):
 
     @pytest.mark.parametrize("pretrained", (True, False))
     def test_pretrained_kw(self, pretrained):
-        @handle_legacy_interface(weights=("pretrained", self.TestWeights.Sentinel))
+        @handle_legacy_interface(weights=("pretrained", self.ModelWeights.Sentinel))
         def builder(*, weights=None):
             pass
 
@@ -242,12 +242,12 @@ def builder(*, weights=None):
     @pytest.mark.parametrize("pretrained", (True, False))
     @pytest.mark.parametrize("positional", (True, False))
     def test_equivalent_behavior_weights(self, pretrained, positional):
-        @handle_legacy_interface(weights=("pretrained", self.TestWeights.Sentinel))
+        @handle_legacy_interface(weights=("pretrained", self.ModelWeights.Sentinel))
         def builder(*, weights=None):
             pass
 
         args, kwargs = ((pretrained,), dict()) if positional else ((), dict(pretrained=pretrained))
-        with pytest.warns(UserWarning, match=f"weights={self.TestWeights.Sentinel if pretrained else None}"):
+        with pytest.warns(UserWarning, match=f"weights={self.ModelWeights.Sentinel if pretrained else None}"):
             builder(*args, **kwargs)
 
     def test_multi_params(self):
@@ -256,7 +256,7 @@ def test_multi_params(self):
 
         @handle_legacy_interface(
             **{
-                weights_param: (pretrained_param, self.TestWeights.Sentinel)
+                weights_param: (pretrained_param, self.ModelWeights.Sentinel)
                 for weights_param, pretrained_param in zip(weights_params, pretrained_params)
             }
         )
@@ -271,7 +271,7 @@ def test_default_callable(self):
         @handle_legacy_interface(
             weights=(
                 "pretrained",
-                lambda kwargs: self.TestWeights.Sentinel if kwargs["flag"] else None,
+                lambda kwargs: self.ModelWeights.Sentinel if kwargs["flag"] else None,
             )
         )
         def builder(*, weights=None, flag):