@@ -13000,7 +13000,7 @@ def test_select_dtypes_include(self):
1300013000 df = DataFrame({'a': list('abc'),
1300113001 'b': list(range(1, 4)),
1300213002 'c': np.arange(3, 6).astype('u1'),
13003- 'd': np.arange(4.0, 7.0),
13003+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1300413004 'e': [True, False, True]})
1300513005 ri = df.select_dtypes(include=[np.number])
1300613006 ei = df[['b', 'c', 'd']]
@@ -13010,7 +13010,7 @@ def test_select_dtypes_exclude(self):
1301013010 df = DataFrame({'a': list('abc'),
1301113011 'b': list(range(1, 4)),
1301213012 'c': np.arange(3, 6).astype('u1'),
13013- 'd': np.arange(4.0, 7.0),
13013+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1301413014 'e': [True, False, True]})
1301513015 re = df.select_dtypes(exclude=[np.number])
1301613016 ee = df[['a', 'e']]
@@ -13020,7 +13020,7 @@ def test_select_dtypes_exclude_include(self):
1302013020 df = DataFrame({'a': list('abc'),
1302113021 'b': list(range(1, 4)),
1302213022 'c': np.arange(3, 6).astype('u1'),
13023- 'd': np.arange(4.0, 7.0),
13023+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1302413024 'e': [True, False, True],
1302513025 'f': pd.date_range('now', periods=3).values})
1302613026 exclude = np.datetime64,
@@ -13030,7 +13030,7 @@ def test_select_dtypes_exclude_include(self):
1303013030 tm.assert_frame_equal(r, e)
1303113031
1303213032 exclude = 'datetime',
13033- include = 'bool', 'int '
13033+ include = 'bool', 'int64', 'int32 '
1303413034 r = df.select_dtypes(include=include, exclude=exclude)
1303513035 e = df[['b', 'e']]
1303613036 tm.assert_frame_equal(r, e)
@@ -13039,7 +13039,7 @@ def test_select_dtypes_not_an_attr_but_still_valid_dtype(self):
1303913039 df = DataFrame({'a': list('abc'),
1304013040 'b': list(range(1, 4)),
1304113041 'c': np.arange(3, 6).astype('u1'),
13042- 'd': np.arange(4.0, 7.0),
13042+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1304313043 'e': [True, False, True],
1304413044 'f': pd.date_range('now', periods=3).values})
1304513045 df['g'] = df.f.diff()
@@ -13071,7 +13071,7 @@ def test_select_dtypes_bad_datetime64(self):
1307113071 df = DataFrame({'a': list('abc'),
1307213072 'b': list(range(1, 4)),
1307313073 'c': np.arange(3, 6).astype('u1'),
13074- 'd': np.arange(4.0, 7.0),
13074+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1307513075 'e': [True, False, True],
1307613076 'f': pd.date_range('now', periods=3).values})
1307713077 with tm.assertRaisesRegexp(ValueError, '.+ is too specific'):
@@ -13085,7 +13085,7 @@ def test_select_dtypes_str_raises(self):
1308513085 'g': list(u('abc')),
1308613086 'b': list(range(1, 4)),
1308713087 'c': np.arange(3, 6).astype('u1'),
13088- 'd': np.arange(4.0, 7.0),
13088+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1308913089 'e': [True, False, True],
1309013090 'f': pd.date_range('now', periods=3).values})
1309113091 string_dtypes = set((str, 'str', np.string_, 'S1',
@@ -13107,7 +13107,7 @@ def test_select_dtypes_bad_arg_raises(self):
1310713107 'g': list(u('abc')),
1310813108 'b': list(range(1, 4)),
1310913109 'c': np.arange(3, 6).astype('u1'),
13110- 'd': np.arange(4.0, 7.0),
13110+ 'd': np.arange(4.0, 7.0, dtype='float64' ),
1311113111 'e': [True, False, True],
1311213112 'f': pd.date_range('now', periods=3).values})
1311313113 with tm.assertRaisesRegexp(TypeError, 'data type.*not understood'):
0 commit comments