Merge pull request #37 from astrofrog/test-fits2bitmap

Test fits2bitmap

imageutils/scripts/fits2bitmap.py

@@ -5,10 +5,12 @@
 import argparse
 from astropy import log
 import astropy.io.fits as fits
-import imageutils
+
 import matplotlib.cm as cm
 import matplotlib.image as mimg
 
+from ..normalization.ui import scale_image
+
 
 def fits2bitmap(filename, exten=0, out_fn=None, scale='linear',
                 power=1.0, noise_level=None, min_cut=None, max_cut=None,
@@ -90,12 +92,12 @@ def fits2bitmap(filename, exten=0, out_fn=None, scale='linear',
                      'name'.format(cmap))
         raise SystemExit()
 
-    image_scaled = imageutils.scale_image(image, scale=scale, power=power,
-                                          noise_level=noise_level,
-                                          min_cut=min_cut, max_cut=max_cut,
-                                          min_percent=min_percent,
-                                          max_percent=max_percent,
-                                          percent=percent)
+    image_scaled = scale_image(image, scale=scale, power=power,
+                               noise_level=noise_level,
+                               min_cut=min_cut, max_cut=max_cut,
+                               min_percent=min_percent,
+                               max_percent=max_percent,
+                               percent=percent)
 
     mimg.imsave(out_fn, image_scaled, cmap=cmap)
     log.info('Saved file to {0}'.format(out_fn))

imageutils/scripts/tests/test_fits2bitmap.py

@@ -0,0 +1,11 @@
+import numpy as np
+from astropy.io import fits
+
+from ..fits2bitmap import fits2bitmap
+
+def test_fits2bitmap_function(tmpdir):
+    filename = tmpdir.join('test.fits').strpath
+    fits.writeto(filename, np.ones((128, 128)))
+    fits2bitmap(filename)
+
+