Merge pull request #228 from lpsinger/six_xrange

Replace xrange with six.moves.xrange to support Python 3

healpy/fitsfunc.py

@@ -24,6 +24,7 @@
 except ImportError:
     import pyfits as pf
 import numpy as np
+import six
 from . import pixelfunc
 from .sphtfunc import Alm
 import warnings
@@ -422,7 +423,7 @@ def mwrfits(filename,data,hdu=1,colnames=None,keys=None):
     else:
         colnames = ['']*len(data)
     cols=[]
-    for line in xrange(len(data)):
+    for line in six.moves.xrange(len(data)):
         cols.append(pf.Column(name=colnames[line],
                                format=getformat(data[line]),
                                array=data[line]))

healpy/projaxes.py

@@ -23,6 +23,7 @@
 import matplotlib
 from matplotlib import axes,ticker,colors,cm,lines,cbook,figure
 import numpy as np
+import six
 from ._healpy_pixel_lib import UNSEEN
 
 pi = np.pi
@@ -394,7 +395,7 @@ def _make_segment(self,x,y,threshold=None):
         elif len(w) >= 2:
             xx.append(x[0:w[0]])
             yy.append(y[0:w[0]])
-            for i in xrange(len(w)-1):
+            for i in six.moves.xrange(len(w)-1):
                 xx.append(x[w[i]:w[i+1]])
                 yy.append(y[w[i]:w[i+1]])
             xx.append(x[w[-1]:])

requirements.txt

@@ -2,3 +2,4 @@ numpy
 cython>=0.16
 matplotlib
 astropy
+six