enforce freqs fed as float in pspecbeam.Jy_to_mK

hera_pspec/pspecbeam.py

@@ -186,8 +186,12 @@ def Jy_to_mK(self, freqs, stokes='pseudo_I'):
         -------
         M : float ndarray, contains Jy -> mK factor at each frequency
         """
-        if isinstance(freqs, (int, np.float, float)):
+        if isinstance(freqs, (np.float, float)):
             freqs = np.array([freqs])
+        elif not isinstance(freqs, np.ndarray):
+            raise TypeError("freqs must be fed as a float ndarray")
+        elif isinstance(freqs, np.ndarray) and freqs.dtype not in (float, np.float, np.float64):
+            raise TypeError("freqs must be fed as a float ndarray")
         if np.min(freqs) < self.beam_freqs.min(): print "Warning: min freq {} < self.beam_freqs.min(), extrapolating...".format(np.min(freqs))
         if np.max(freqs) > self.beam_freqs.max(): print "Warning: max freq {} > self.beam_freqs.max(), extrapolating...".format(np.max(freqs))
 

hera_pspec/tests/test_pspecbeam.py

@@ -80,7 +80,11 @@ def test_UVbeam(self):
         nt.assert_equal(len(M), 11)
         nt.assert_almost_equal(M[0], 41.360105524572283)
         M = self.bm.Jy_to_mK(99e6)
-        M = self.bm.Jy_to_mK(201)
+        M = self.bm.Jy_to_mK(201e6)
+        # test exception
+        nt.assert_raises(TypeError, self.bm.Jy_to_mK, [1])
+        nt.assert_raises(TypeError, self.bm.Jy_to_mK, np.array([1]))
+
 
     def test_Gaussbeam(self):
         Om_p = self.gauss.power_beam_int()