wip

Author: hayd

pandas_data_readers/tests/test_data.py

@@ -9,8 +9,13 @@
 import numpy as np
 import pandas as pd
 from pandas import DataFrame, Timestamp
-from pandas.util.testing import (assert_series_equal, assert_produces_warning,
-                                 network, assert_frame_equal)
+from pandas.util.testing import (assert_frame_equal, assert_series_equal,
+                                 network)
+try:
+    from pandas.util.testing import assert_produces_warning
+except ImportError:
+    assert_produces_warning = None
+
 import pandas.util.testing as tm
 from numpy.testing import assert_array_equal
 
@@ -401,6 +406,9 @@ def tearDownClass(cls):
 
     @network
     def test_options_source_warning(self):
+        if not assert_produces_warning:
+            raise SkipTest("old version of pandas without "
+                           "compat.assert_produces_warning")
         with assert_produces_warning():
             aapl = web.Options('aapl')
 

pandas_data_readers/tests/test_wb.py

@@ -1,14 +1,24 @@
 import nose
 
 import pandas
-from pandas.compat import u
 from pandas.util.testing import network
 from pandas.util.testing import assert_frame_equal
 from numpy.testing.decorators import slow
 import pandas.util.testing as tm
 
 from pandas_data_readers.wb import search, download, get_countries
 
+try:
+    from pandas.compat import u
+except ImportError:
+    try:
+        unicode # python 2
+        def u(s):
+            return unicode(s, "unicode_escape")
+    except NameError:
+        def u(s):
+            return s
+
 
 class TestWB(tm.TestCase):