prefix test model class with _

spacetelescope · Nov 20, 2024 · 6b7fad7 · 6b7fad7
1 parent a6b4a90
commit 6b7fad7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/stdatamodels/jwst/datamodels/tests/test_models.py b/src/stdatamodels/jwst/datamodels/tests/test_models.py
@@ -129,15 +129,15 @@ def test_imagemodel():
 
 
 def test_model_with_nonstandard_primary_array():
-    class NonstandardPrimaryArrayModel(JwstDataModel):
+    class _NonstandardPrimaryArrayModel(JwstDataModel):
         schema_url = os.path.join(ROOT_DIR, "nonstandard_primary_array.schema.yaml")
 
         # The wavelength array is the primary array.
         # Try commenting this function out and the problem goes away.
         def get_primary_array_name(self):
             return 'wavelength'
 
-    m = NonstandardPrimaryArrayModel((10,))
+    m = _NonstandardPrimaryArrayModel((10,))
     assert 'wavelength' in list(m.keys())
     assert m.wavelength.sum() == 0