Fixed codestyle issues.

Author: hpparvi

specreduce/tests/test_align_along_trace.py

@@ -18,19 +18,19 @@ def mk_test_image():
 def test_align_spectrum_along_trace_bad_input():
     image, trace = mk_test_image()
     with pytest.raises(ValueError, match='Unre'):
-        im = align_spectrum_along_trace(image, None)
+        im = align_spectrum_along_trace(image, None)   # noqa
 
     with pytest.raises(ValueError, match='method must be'):
-        im = align_spectrum_along_trace(image, trace, method='int')
+        im = align_spectrum_along_trace(image, trace, method='int')   # noqa
 
     with pytest.raises(ValueError, match='Spectral axis length'):
-        im = align_spectrum_along_trace(image.T, trace, method='interpolate', disp_axis=0)
+        im = align_spectrum_along_trace(image.T, trace, method='interpolate', disp_axis=0)   # noqa
 
     with pytest.raises(ValueError, match='Displacement axis must be'):
-        im = align_spectrum_along_trace(image, trace, disp_axis=2)
+        im = align_spectrum_along_trace(image, trace, disp_axis=2)  # noqa
 
     with pytest.raises(ValueError, match='The number of image dimensions must be'):
-        im = align_spectrum_along_trace(np.zeros((3,6,9)), trace)
+        im = align_spectrum_along_trace(np.zeros((3, 6, 9)), trace)  # noqa
 
 
 @pytest.mark.parametrize("method, truth_data, truth_mask, truth_ucty",
@@ -39,18 +39,18 @@ def test_align_spectrum_along_trace_bad_input():
                                      [0, 0, 0, 0.25, 0.5, 0.25, 0, 0, 0]]).T,
                            np.array([[0, 0, 0, 0, 0, 0, 0, 1, 1],
                                      [1, 1, 0, 0, 0, 0, 0, 0, 0]]).astype(bool).T,
-                           np.array([[1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. , 1. ],
+                           np.array([[1., 1., 1., 1., 1., 1., 1., 1., 1.],
                                      [0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]]).T),
                           ('shift',
-                           np.array([[0. , 0. , 0. , 0. , 1. , 0. , 0. , 0. , 0. ],
-                                     [0. , 0. , 0. , 0.5, 0.5, 0. , 0. , 0. , 0. ]]).T,
+                           np.array([[0., 0., 0., 0., 1., 0., 0., 0., 0.],
+                                     [0., 0., 0., 0.5, 0.5, 0., 0., 0., 0.]]).T,
                            np.array([[0, 0, 0, 0, 0, 0, 0, 0, 1],
                                      [1, 0, 0, 0, 0, 0, 0, 0, 0]]).astype(bool).T,
-                           np.ones((9,2)))],
+                           np.ones((9, 2)))],
                          ids=('method=interpolate', 'method=shift'))
 def test_align_spectrum_along_trace(method, truth_data, truth_mask, truth_ucty):
     image, trace = mk_test_image()
-    im =  align_spectrum_along_trace(image, trace, method=method)
+    im = align_spectrum_along_trace(image, trace, method=method)
     assert im.shape == image.shape
     assert im.unit == u.DN
     assert im.uncertainty.uncertainty_type == 'var'

specreduce/utils/utils.py

@@ -9,7 +9,8 @@
 from specreduce.tracing import Trace, FlatTrace
 from specreduce.extract import _ap_weight_image, _align_along_trace
 
-__all__ = ['measure_cross_dispersion_profile', '_align_along_trace', 'align_2d_spectrum_along_trace']
+__all__ = ['measure_cross_dispersion_profile', '_align_along_trace',
+           'align_2d_spectrum_along_trace']
 
 
 def _get_image_ndim(image):