Merge pull request #385 from spacetelescope/0.17.1x

0.17.1x
spacetelescope · Nov 27, 2021 · e8a5ecf · e8a5ecf
2 parents 55d5b8f + e747e34
commit e8a5ecf
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,3 +1,12 @@
+0.17.1 (2021-11-27)
+-------------------
+
+Bug Fixes
+^^^^^^^^^
+
+- Fixed a bug with StokesProfile and array types. [#384]
+
+
 0.17.0 (2021-11-17)
 -------------------
 Bug Fixes

diff --git a/gwcs/coordinate_frames.py b/gwcs/coordinate_frames.py
@@ -731,11 +731,11 @@ def from_index(cls, indexes):
         """
 
         nans = np.isnan(indexes)
-        indexes = np.asanyarray(indexes, dtype=int)
+        indexes = np.asarray(indexes, dtype=int)
         out = np.empty_like(indexes, dtype=object)
 
         for profile, index in cls.profiles.items():
-            out[indexes == index] = profile
+            out[indexes == index] = cls(profile)
 
         out[nans] = np.nan
 

diff --git a/gwcs/tests/test_coordinate_systems.py b/gwcs/tests/test_coordinate_systems.py
@@ -286,6 +286,9 @@ def test_stokes_frame():
     assert sf.coordinate_to_quantity('I') == 0 * u.one
     assert sf.coordinate_to_quantity(0) == 0
 
+def test_stokes_profile():
+    assert (cf.StokesProfile.from_index(np.arange(-8, 4) * u.one) == np.array(['YX', 'XY', 'YY,', 'XX', 'LR', 'RL', 'LL', 'RR', 'I', 'Q', 'U', 'V'], dtype="U2")).all()
+
 
 @pytest.mark.parametrize('inp', [
     (211 * u.AA, 0 * u.s, 0 * u.one, 0 * u.one),