cleanup

qpformat/file_formats/single_npy_numpy.py

@@ -8,11 +8,12 @@
 
 class SingleNpyNumpy(DataSet):
     """Field data stored in numpy's .npy file format
-    
+
     The experimental data given in `path` consist of a single
     2D ndarray (no pickled objects). The ndarray is either
     complex-valued (scattered field) or real-valued (phase).
     """
+
     def __len__(self):
         return 1
 

tests/test_single_npy_numpy.py

@@ -1,4 +1,4 @@
-from os.path import abspath, dirname, join
+from os.path import abspath, dirname
 import sys
 import tempfile
 
@@ -13,9 +13,9 @@ def test_load_phase():
     # generate test data
     tf = tempfile.mktemp(suffix=".npy", prefix="qpformat_test_")
     phase = np.ones((20, 20), dtype=float)
-    phase *= np.linspace(0, .3, 20).reshape(-1,1) 
+    phase *= np.linspace(0, .3, 20).reshape(-1, 1)
     np.save(tf, phase)
-    
+
     ds = qpformat.load_data(tf)
     assert np.allclose(ds.get_qpimage().pha, phase, atol=1e-15, rtol=0)
     assert ds.path == tf
@@ -26,12 +26,12 @@ def test_load_field():
     # generate test data
     tf = tempfile.mktemp(suffix=".npy", prefix="qpformat_test_")
     phase = np.ones((20, 20), dtype=float)
-    phase *= np.linspace(0, .3, 20).reshape(-1,1) 
+    phase *= np.linspace(0, .3, 20).reshape(-1, 1)
     amplitude = np.ones((20, 20), dtype=float)
-    amplitude += np.linspace(-.1, .12, 20).reshape(1,-1)
+    amplitude += np.linspace(-.1, .12, 20).reshape(1, -1)
     field = amplitude * np.exp(1j * phase)
     np.save(tf, field)
-    
+
     ds = qpformat.load_data(tf)
     assert np.allclose(ds.get_qpimage().pha, phase, atol=1e-15, rtol=0)
     assert np.allclose(ds.get_qpimage().amp, amplitude, atol=1e-15, rtol=0)