Fix tests

mattip · Mar 22, 2017 · 52f9008 · 52f9008
1 parent 41008c7
commit 52f9008
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pandas/tests/scalar/test_timestamp.py b/pandas/tests/scalar/test_timestamp.py
@@ -604,7 +604,7 @@ def test_nat_vector_field_access(self):
         for field in fields:
             result = getattr(idx, field)
             expected = [getattr(x, field) for x in idx]
-            self.assert_numpy_array_equal(result, np.array(expected))
+            self.assert_index_equal(result, pd.Index(expected))
 
         s = pd.Series(idx)
 

diff --git a/pandas/tests/tools/test_pivot.py b/pandas/tests/tools/test_pivot.py
@@ -1367,7 +1367,7 @@ def test_daily(self):
         with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
             annual = pivot_annual(ts, 'D')
 
-        doy = ts.index.dayofyear
+        doy = np.asarray(ts.index.dayofyear)
 
         with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
             doy[(~isleapyear(ts.index.year)) & (doy >= 60)] += 1

diff --git a/pandas/tests/tseries/test_timezones.py b/pandas/tests/tseries/test_timezones.py
@@ -358,7 +358,7 @@ def test_field_access_localize(self):
         dr = date_range('2011-10-02 00:00', freq='h', periods=10,
                         tz=self.tzstr('America/Atikokan'))
 
-        expected = pd.Index(np.arange(10, dtype=np.int64))
+        expected = Index(np.arange(10, dtype=np.int64))
         self.assert_index_equal(dr.hour, expected)
 
     def test_with_tz(self):