Update test_core.py

punch-mission · Nov 2, 2023 · e3736b9 · e3736b9
1 parent 77b0640
commit e3736b9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -27,9 +27,9 @@
 @fixture
 def npol_ones():
     data_out = []
-    data_out.append(("angle_1", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': 60})))
-    data_out.append(("angle_2", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': 0})))
-    data_out.append(("angle_3", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': -60})))
+    data_out.append(("angle_1", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': 60, 'OBSRVTRY': 'LASCO'})))
+    data_out.append(("angle_2", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': 0, 'OBSRVTRY': 'LASCO'})))
+    data_out.append(("angle_3", NDCube(np.array([1]), wcs=wcs, meta={'POLAR': -60, 'OBSRVTRY': 'LASCO'})))
     # data_out.append(("alpha", NDCube(np.array([0])*u.radian, wcs=wcs)))
     return NDCollection(data_out, meta={}, aligned_axes="all")
 
@@ -142,7 +142,7 @@ def test_determine_input_kind_npol(npol_ones):
     assert determine_input_kind(npol_ones), "npol"
 
 def test_determine_input_kind_fourpol(four_pol_ones):
-    assert determine_input_kind(four_pol_ones), "npol"
+    assert determine_input_kind(four_pol_ones), "fourpol"
 
 def test_determine_input_kind_mzp(mzp_ones):
     assert determine_input_kind(mzp_ones), "MZP"
@@ -441,4 +441,4 @@ def test_bp3_to_bthp(bp3_ones):
 #     """test pB from single angle fn in core"""
 #
 #     output=pB_from_single_angle(B, B_theta, theta, alpha)
-#     np.testing.assert_allclose(output, expected, rtol=1e-05)
+#     np.testing.assert_allclose(output, expected, rtol=1e-05)