TST: clean warnings

Author: jreback

Makefile

@@ -12,6 +12,9 @@ clean_pyc:
 build: clean_pyc
 	python setup.py build_ext --inplace
 
+lint-diff:
+	git diff master --name-only -- "*.py" | grep "pandas" | xargs flake8
+
 develop: build
 	-python setup.py develop
 

pandas/tests/indexing/common.py

@@ -127,11 +127,11 @@ def get_result(self, obj, method, key, axis):
             key = obj._get_axis(axis)[key]
 
         # in case we actually want 0 index slicing
-        try:
-            with catch_warnings(record=True):
+        with catch_warnings(record=True):
+            try:
                 xp = getattr(obj, method).__getitem__(_axify(obj, key, axis))
-        except:
-            xp = getattr(obj, method).__getitem__(key)
+            except:
+                xp = getattr(obj, method).__getitem__(key)
 
         return xp
 

pandas/tests/indexing/test_floats.py

@@ -169,19 +169,31 @@ def f():
 
         # mixed index so we have label
         # indexing
-        for idxr in [lambda x: x.ix,
-                     lambda x: x]:
+        for idxr in [lambda x: x]:
 
             def f():
-                with catch_warnings(record=True):
-                    idxr(s3)[1.0]
+                idxr(s3)[1.0]
 
             pytest.raises(TypeError, f)
 
             result = idxr(s3)[1]
             expected = 2
             assert result == expected
 
+        # mixed index so we have label
+        # indexing
+        for idxr in [lambda x: x.ix]:
+            with catch_warnings(record=True):
+
+                def f():
+                    idxr(s3)[1.0]
+
+                pytest.raises(TypeError, f)
+
+                result = idxr(s3)[1]
+                expected = 2
+                assert result == expected
+
         pytest.raises(TypeError, lambda: s3.iloc[1.0])
         pytest.raises(KeyError, lambda: s3.loc[1.0])
 
@@ -479,16 +491,14 @@ def test_slice_integer_frame_getitem(self):
             index = index(5)
             s = DataFrame(np.random.randn(5, 2), index=index)
 
-            for idxr in [lambda x: x.loc,
-                         lambda x: x.ix]:
+            def f(idxr):
 
                 # getitem
                 for l in [slice(0.0, 1),
                           slice(0, 1.0),
                           slice(0.0, 1.0)]:
 
-                    with catch_warnings(record=True):
-                        result = idxr(s)[l]
+                    result = idxr(s)[l]
                     indexer = slice(0, 2)
                     self.check(result, s, indexer, False)
 
@@ -516,8 +526,7 @@ def f():
                                (slice(0, 0.5), slice(0, 1)),
                                (slice(0.5, 1.5), slice(1, 2))]:
 
-                    with catch_warnings(record=True):
-                        result = idxr(s)[l]
+                    result = idxr(s)[l]
                     self.check(result, s, res, False)
 
                     # positional indexing
@@ -532,9 +541,8 @@ def f():
                           slice(3.0, 4.0)]:
 
                     sc = s.copy()
-                    with catch_warnings(record=True):
-                        idxr(sc)[l] = 0
-                        result = idxr(sc)[l].values.ravel()
+                    idxr(sc)[l] = 0
+                    result = idxr(sc)[l].values.ravel()
                     assert (result == 0).all()
 
                     # positional indexing
@@ -543,6 +551,10 @@ def f():
 
                     pytest.raises(TypeError, f)
 
+            f(lambda x: x.loc)
+            with catch_warnings(record=True):
+                f(lambda x: x.ix)
+
     def test_slice_float(self):
 
         # same as above, but for floats

pandas/tests/indexing/test_indexing.py

@@ -884,8 +884,10 @@ def test_partial_boolean_frame_indexing(self):
 
     def test_no_reference_cycle(self):
         df = pd.DataFrame({'a': [0, 1], 'b': [2, 3]})
-        for name in ('loc', 'iloc', 'ix', 'at', 'iat'):
+        for name in ('loc', 'iloc', 'at', 'iat'):
             getattr(df, name)
+        with catch_warnings(record=True):
+            getattr(df, 'ix')
         wr = weakref.ref(df)
         del df
         assert wr() is None