Adding back the capitalization test

Author: datumbox

test/test_prototype_models.py

@@ -117,6 +117,7 @@ def test_schema_meta_validation(model_fn):
 
     problematic_weights = {}
     incorrect_params = []
+    bad_names = []
     for w in weights_enum:
         missing_fields = fields - set(w.meta.keys())
         if missing_fields:
@@ -133,9 +134,12 @@ def test_schema_meta_validation(model_fn):
         else:
             if w.meta.get("num_params") != weights_enum.DEFAULT.meta.get("num_params"):
                 incorrect_params.append(w)
+        if not w.name.isupper():
+            bad_names.append(w)
 
     assert not problematic_weights
     assert not incorrect_params
+    assert not bad_names
 
 
 @pytest.mark.parametrize("model_fn", TM.get_models_from_module(models))