Full step test

romancal/assign_wcs/tests/test_wcs.py

@@ -1,11 +1,12 @@
+import numpy as np
+import os
 import pytest
 
-import numpy as np
 from numpy.testing import assert_allclose
 
 from gwcs.wcstools import grid_from_bounding_box
 
-from romancal.assign_wcs.assign_wcs_step import load_wcs
+from romancal.assign_wcs.assign_wcs_step import AssignWcsStep
 from roman_datamodels import datamodels as rdm
 from roman_datamodels.testing import utils as testutil
 
@@ -14,12 +15,14 @@
 
 def create_image():
     l2 = testutil.mk_level2_image()
+
     l2.meta.wcsinfo.v2_ref = -503
     l2.meta.wcsinfo.v3_ref = -318
     l2.meta.wcsinfo.ra_ref = 156
     l2.meta.wcsinfo.dec_ref = 54.2
     l2.meta.wcsinfo.vparity = -1
     l2.meta.wcsinfo.roll_ref = 0.15
+
     l2im = rdm.ImageModel(l2)
     return l2im
 
@@ -36,9 +39,17 @@ def create_distortion():
 
 
 @pytest.mark.parametrize("distortion", create_distortion())
-def test_wcs(distortion):
+@pytest.mark.skipif(
+    os.environ.get("CI") == "true",
+    reason="Roman CRDS servers are not currently available outside the internal network"
+)
+def test_wcs(tmpdir, distortion):
+    file_name = str(tmpdir / 'distortion.asdf')
+    dist = rdm.DistortionRefModel(distortion)
+    dist.save(file_name)
+
     l2im = create_image()
-    l2_wcs = load_wcs(l2im, {'distortion': distortion})
+    l2_wcs = AssignWcsStep.call(l2im, override_distortion=file_name)
 
     assert l2_wcs.meta.wcs is not None
     assert l2_wcs.meta.cal_step.assign_wcs == 'COMPLETE'