NIST: added ability to query of multiple linenames

Author: nkphysics

astroquery/nist/core.py

@@ -85,7 +85,13 @@ def _args_to_payload(self, *args, **kwargs):
 
         """
         request_payload = {}
-        request_payload["spectra"] = kwargs['linename']
+        linename = kwargs["linename"]
+        if isinstance(linename, str):
+            request_payload["spectra"] = linename
+        elif isinstance(linename, list):
+            request_payload["spectra"] = " ".join(linename)
+        else:
+            raise TypeError("linename must str or list")
         (min_wav, max_wav, wav_unit) = _parse_wavelength(args[0], args[1])
         request_payload["low_wl"] = min_wav
         request_payload["upp_wl"] = max_wav

astroquery/nist/tests/test_nist.py

@@ -45,6 +45,9 @@ def test_query_async(patch_get):
                                           linename="H I", get_query_payload=True)
     assert response['spectra'] == "H I"
     assert response['unit'] == nist.core.Nist.unit_code['nm']
+    response = nist.core.Nist.query_async(4000 * u.nm, 7000 * u.nm,
+                                          linename=["H I", "Fe I"], get_query_payload=True)
+    assert response["spectra"] == "H I Fe I"
     response = nist.core.Nist.query_async(4000 * u.nm, 7000 * u.nm, linename="H I")
     assert response is not None