17: Check input in kplot

pymule/plot.py

@@ -57,6 +57,15 @@ def kplot(sigma, k, sigmar=[], kr=[],
           labelk="$K$", labelkr=None,
           labelsigma=None, labelsigmar=None):
 
+    if type(sigma) == np.ndarray:
+        sigma = [sigma]
+    if type(sigmar) == np.ndarray:
+        sigmar = [sigmar]
+    if type(k) == np.ndarray:
+        k = [k]
+    if type(kr) == np.ndarray:
+        kr = [kr]
+
     if labelsigma is None:
         labelsigma = r"$\D\sigma\,/\,\D " + labelx + "$"