fix: test cases

- fixed flux array length check so that the 2D mwm visits are checked properly

specutils/io/default_loaders/tests/test_sdss_v.py

@@ -438,7 +438,7 @@ def spec_HDUList(n_spectra):
     [
         ("mwm-temp", None, None, False, [0, 0, 1, 0], 1),  # visit
         ("mwm-temp", 3, None, False, [0, 0, 1, 0], 1),
-        ("mwm-temp", None, 2, False, [0, 1, 1, 0], 3),  # multi-ext visits
+        ("mwm-temp", None, 2, False, [0, 1, 1, 0], 2),  # multi-ext visits
         ("mwm-temp", 2, 2, False, [0, 1, 1, 0], 3),
         ("mwm-temp", None, None, True, [0, 0, 1, 0], 1),  # star
         ("mwm-temp", 3, None, True, [0, 0, 1, 0], 1),
@@ -463,7 +463,10 @@ def test_mwm_1d(file_obj, hdu, visit, with_wl, hduflags, nvisits):
         raise ValueError(
             "INSTRMNT tag in test HDU header is not set properly.")
     assert len(data.spectral_axis.value) == length
-    assert len(data.flux.value) == length
+    assert data.flux.value.shape[-1] == length
+    if nvisits > 1:
+        assert data.flux.value.shape[0] == nvisits
+
     assert data.spectral_axis.unit == Angstrom
     assert data.flux.unit == Unit("1e-17 erg / (s cm2 Angstrom)")
     os.remove(tmpfile)
@@ -485,8 +488,9 @@ def test_mwm_1d(file_obj, hdu, visit, with_wl, hduflags, nvisits):
 def test_mwm_list(file_obj, with_wl, hduflags):
     """Test mwm SpectrumList loader"""
     tmpfile = str(file_obj) + ".fits"
-    mwm_HDUList(hduflags, with_wl,
-                nvisits=1 if with_wl else 3).writeto(tmpfile, overwrite=True)
+    nvisits = 1 if with_wl else 3
+    mwm_HDUList(hduflags, with_wl, nvisits=nvisits).writeto(tmpfile,
+                                                            overwrite=True)
 
     data = SpectrumList.read(tmpfile)
     assert isinstance(data, SpectrumList)
@@ -505,7 +509,9 @@ def test_mwm_list(file_obj, with_wl, hduflags):
         else:
             assert data[i].meta['datatype'].lower() == 'mwmvisit'
         assert len(data[i].spectral_axis.value) == length
-        assert len(data[i].flux.value) == length
+        assert data[i].flux.value.shape[-1] == length
+        if nvisits > 1:
+            assert data[i].flux.value.shape[0] == nvisits
         assert data[i].spectral_axis.unit == Angstrom
         assert data[i].flux.unit == Unit("1e-17 erg / (s cm2 Angstrom)")
     os.remove(tmpfile)